Fix tests

This commit is contained in:
Reuben Morais
2019-08-29 12:18:33 +02:00
parent 3c5aeb5e22
commit 28644a717e
3 changed files with 4 additions and 2 deletions
+1 -1
View File
@@ -250,7 +250,7 @@ def set_init_dict(model_dict, checkpoint, c):
def setup_model(num_chars, num_speakers, c):
print(" > Using model: {}".format(c.model))
MyModel = importlib.import_module('models.' + c.model.lower())
MyModel = importlib.import_module('TTS.models.' + c.model.lower())
MyModel = getattr(MyModel, c.model)
if c.model.lower() in ["tacotron", "tacotrongst"]:
model = MyModel(