Medix-AI/coqui-tts
Medix-AI
/
coqui-tts
mirror of https://github.com/coqui-ai/TTS.git
1
0
Fork 0
Code Issues Packages Projects Releases Wiki Activity

Allow choosing phonemizer

This commit is contained in:
Eren Gölge 2021-11-24 18:37:54 +01:00
parent 22f0c58fe1
commit 4e83bf3968
1 changed files with 9 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

10
TTS/tts/utils/text/tokenizer.py
View File

@ -117,14 +117,22 @@ class TTSTokenizer:
Args: Args:
config (Coqpit): Coqpit model config. config (Coqpit): Coqpit model config.
""" """
# init cleaners
if isinstance(config.text_cleaner, (str, list)): if isinstance(config.text_cleaner, (str, list)):
text_cleaner = getattr(cleaners, config.text_cleaner) text_cleaner = getattr(cleaners, config.text_cleaner)
if config.use_phonemes: if config.use_phonemes:
# init phoneme set
characters = IPAPhonemes().init_from_config(config) characters = IPAPhonemes().init_from_config(config)
phonemizer_kwargs = {"language": config.phoneme_language} phonemizer_kwargs = {"language": config.phoneme_language}
phonemizer = get_phonemizer_by_name(DEF_LANG_TO_PHONEMIZER[config.phoneme_language], **phonemizer_kwargs)
# init phonemizer
if "phonemizer" in config and config.phonemizer:
phonemizer = get_phonemizer_by_name(config.phonemizer, **phonemizer_kwargs)
else:
phonemizer = get_phonemizer_by_name(DEF_LANG_TO_PHONEMIZER[config.phoneme_language], **phonemizer_kwargs)
else: else:
# init character set
characters = Graphemes().init_from_config(config) characters = Graphemes().init_from_config(config)
return TTSTokenizer( return TTSTokenizer(
config.use_phonemes, text_cleaner, characters, phonemizer, config.add_blank, config.enable_eos_bos_chars config.use_phonemes, text_cleaner, characters, phonemizer, config.add_blank, config.enable_eos_bos_chars