coqui-ai
/
TTS
mirror of https://github.com/coqui-ai/TTS.git
1
0
Fork 0
Code Issues Packages Projects Releases Wiki Activity

fix the glow-tts in setup_model

pull/441/head
Eren Gölge 2021-04-21 13:10:40 +02:00
parent 8764d02eb2
commit 09890c7421
1 changed files with 2 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

3
TTS/tts/utils/generic_utils.py
View File

@ -119,6 +119,7 @@ def setup_model(num_chars, num_speakers, c, speaker_embedding_dim=None):
encoder_type=c.encoder_type, encoder_type=c.encoder_type,
encoder_params=c.encoder_params, encoder_params=c.encoder_params,
use_encoder_prenet=c["use_encoder_prenet"], use_encoder_prenet=c["use_encoder_prenet"],
inference_noise_scale=c.get("inference_noise_scale", 0.33),
num_flow_blocks_dec=12, num_flow_blocks_dec=12,
kernel_size_dec=5, kernel_size_dec=5,
dilation_rate=1, dilation_rate=1,
@ -130,7 +131,7 @@ def setup_model(num_chars, num_speakers, c, speaker_embedding_dim=None):
num_squeeze=2, num_squeeze=2,
sigmoid_scale=False, sigmoid_scale=False,
mean_only=True, mean_only=True,
external_speaker_embedding_dim=speaker_embedding_dim, speaker_embedding_dim=speaker_embedding_dim,
) )
elif c.model.lower() == "speedy_speech": elif c.model.lower() == "speedy_speech":
model = MyModel( model = MyModel(