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

Fix Tacotron num_char init

This commit is contained in:
Eren Gölge 2021-08-10 08:56:34 +00:00
parent 7eb94f760b
commit c8b9ca3d71
2 changed files with 6 additions and 4 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

4
TTS/tts/models/tacotron.py
View File

@ -23,8 +23,8 @@ class Tacotron(BaseTacotron):
def __init__(self, config: Coqpit): def __init__(self, config: Coqpit):
super().__init__(config) super().__init__(config)
self.num_chars, self.config = self.get_characters(config) chars, self.config = self.get_characters(config)
config.num_chars = self.num_chars config.num_chars = self.num_chars = len(chars)
# pass all config fields to `self` # pass all config fields to `self`
# for fewer code change # for fewer code change

6
TTS/vocoder/models/__init__.py
View File

@ -24,8 +24,10 @@ def setup_model(config: Coqpit):
elif config.model.lower() == "wavegrad": elif config.model.lower() == "wavegrad":
MyModel = getattr(MyModel, "Wavegrad") MyModel = getattr(MyModel, "Wavegrad")
else: else:
MyModel = getattr(MyModel, to_camel(config.model)) try:
raise ValueError(f"Model {config.model} not exist!") MyModel = getattr(MyModel, to_camel(config.model))
except ModuleNotFoundError as e:
raise ValueError(f"Model {config.model} not exist!") from e
model = MyModel(config) model = MyModel(config)
return model return model