Mirrors/TTS
1
0
Fork 0
mirror of https://github.com/coqui-ai/TTS.git synced 2025-12-25 12:49:29 +01:00
Code Issues Packages Projects Releases Wiki Activity

Fix tokenizer for punc only (#1717)

Browse Source
This commit is contained in:
WeberJulian
2022-07-06 22:59:41 +02:00
committed by GitHub
parent 9e00e31e37
commit 5cef6facb0
2 changed files with 8 additions and 1 deletions
Download Patch File Download Diff File Expand all files Collapse all files

2
TTS/tts/utils/text/punctuation.py
View File

@@ -137,7 +137,7 @@ class Punctuation:
# nothing have been phonemized, returns the puncs alone
if not text:
return ["".join(m.mark for m in puncs)]
return ["".join(m.punc for m in puncs)]
current = puncs[0]

7
tests/text_tests/test_tokenizer.py
View File

@@ -30,6 +30,13 @@ class TestTTSTokenizer(unittest.TestCase):
test_hat = self.tokenizer_ph.ids_to_text(ids)
self.assertEqual(text_ph, test_hat)
def test_text_to_ids_phonemes_punctuation(self):
text = "..."
text_ph = self.ph.phonemize(text, separator="")
ids = self.tokenizer_ph.text_to_ids(text)
test_hat = self.tokenizer_ph.ids_to_text(ids)
self.assertEqual(text_ph, test_hat)
def test_text_to_ids_phonemes_with_eos_bos(self):
text = "Bu bir Örnek."
self.tokenizer_ph.use_eos_bos = True