wassname/TTS
1
0 Fork 0
mirror of https://github.com/wassname/TTS.git synced 2026-10-27 12:30:27 +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 authored and Eren Gölge committed 2021-03-08 02:54:21 +01:00
1 parent 5e2bc8c99f
commit ed56944c4a
1 file changed
+3 -3
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'],