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

Update `tts.setup_model` 

Run `model.make_symbols()` if availabe to set the symbol list
This commit is contained in:
Eren Gölge 2021-08-07 21:50:06 +00:00
parent f5a6aa974f
commit 232a5abb6a
1 changed files with 7 additions and 4 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

11
TTS/tts/models/__init__.py
View File

@ -4,20 +4,23 @@ from TTS.utils.generic_utils import find_module
def setup_model(config): def setup_model(config):
print(" > Using model: {}".format(config.model)) print(" > Using model: {}".format(config.model))
MyModel = find_module("TTS.tts.models", config.model.lower()) MyModel = find_module("TTS.tts.models", config.model.lower())
# define set of characters used by the model # define set of characters used by the model
if config.characters is not None: if config.characters is not None:
# set characters from config # set characters from config
symbols, phonemes = make_symbols(**config.characters.to_dict()) # pylint: disable=redefined-outer-name if hasattr(MyModel, "make_symbols"):
symbols = MyModel.make_symbols(config)
else:
symbols, phonemes = make_symbols(**config.characters)
else: else:
from TTS.tts.utils.text.symbols import phonemes, symbols # pylint: disable=import-outside-toplevel from TTS.tts.utils.text.symbols import phonemes, symbols # pylint: disable=import-outside-toplevel
if config.use_phonemes:
symbols = phonemes
# use default characters and assign them to config # use default characters and assign them to config
config.characters = parse_symbols() config.characters = parse_symbols()
num_chars = len(phonemes) if config.use_phonemes else len(symbols)
# consider special `blank` character if `add_blank` is set True # consider special `blank` character if `add_blank` is set True
num_chars = num_chars + getattr(config, "add_blank", False) num_chars = len(symbols) + getattr(config, "add_blank", False)
config.num_chars = num_chars config.num_chars = num_chars
# compatibility fix # compatibility fix
if "model_params" in config: if "model_params" in config: