wassname/TTS
1
0
Fork 0
mirror of https://github.com/wassname/TTS.git synced 2026-09-09 11:16:00 +08:00
Code Issues Packages Projects Releases Wiki Activity

improve robustness of defining wavernn in config file

Browse Source
This commit is contained in:
Branislav Gerazov
2021-03-08 02:54:21 +01:00
committed by Eren Gölge
parent 5e2bc8c99f
commit ed56944c4a
1 changed files with 3 additions and 3 deletions
Download Patch File Download Diff File Expand all files Collapse all files
TTS/vocoder/utils/generic_utils.py
+3 -3
View File
@@ -71,10 +71,10 @@ def setup_generator(c):
MyModel = importlib.import_module('TTS.vocoder.models.' +
c.generator_model.lower())
# this is to preserve the WaveRNN class name (instead of Wavernn)
if c.generator_model != 'WaveRNN':
MyModel = getattr(MyModel, to_camel(c.generator_model))
if c.generator_model.lower() == 'wavernn':
MyModel = getattr(MyModel, 'WaveRNN')
else:
MyModel = getattr(MyModel, c.generator_model)
MyModel = getattr(MyModel, to_camel(c.generator_model))
if c.generator_model.lower() in 'wavernn':
model = MyModel(
rnn_dims=c.wavernn_model_params['rnn_dims'],