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

improve robustness of defining wavernn in config file

This commit is contained in:
Branislav Gerazov 2021-02-05 13:26:33 +01:00
parent 24ffa9e9f6
commit f063545325
1 changed files with 3 additions and 3 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

6
TTS/vocoder/utils/generic_utils.py
View File

@ -71,10 +71,10 @@ def setup_generator(c):
MyModel = importlib.import_module('TTS.vocoder.models.' + MyModel = importlib.import_module('TTS.vocoder.models.' +
c.generator_model.lower()) c.generator_model.lower())
# this is to preserve the WaveRNN class name (instead of Wavernn) # this is to preserve the WaveRNN class name (instead of Wavernn)
if c.generator_model != 'WaveRNN': if c.generator_model.lower() == 'wavernn':
MyModel = getattr(MyModel, to_camel(c.generator_model)) MyModel = getattr(MyModel, 'WaveRNN')
else: else:
MyModel = getattr(MyModel, c.generator_model) MyModel = getattr(MyModel, to_camel(c.generator_model))
if c.generator_model.lower() in 'wavernn': if c.generator_model.lower() in 'wavernn':
model = MyModel( model = MyModel(
rnn_dims=c.wavernn_model_params['rnn_dims'], rnn_dims=c.wavernn_model_params['rnn_dims'],