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

update argumnet check for non tacotron models

This commit is contained in:
erogol 2020-12-21 12:29:07 +01:00
parent 27a75de15f
commit 57ef53bef3
1 changed files with 9 additions and 3 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

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

@ -126,15 +126,21 @@ def setup_model(num_chars, num_speakers, c, speaker_embedding_dim=None):
hidden_channels_enc=192, hidden_channels_enc=192,
hidden_channels_dec=192, hidden_channels_dec=192,
use_encoder_prenet=True, use_encoder_prenet=True,
rel_attn_window_size=4,
external_speaker_embedding_dim=speaker_embedding_dim) external_speaker_embedding_dim=speaker_embedding_dim)
elif c.model.lower() == "speedy_speech":
model = MyModel(num_chars=num_chars + getattr(c, "add_blank", False),
out_channels=c.audio['num_mels'],
hidden_channels=128,
encoder_type=c['encoder_type'],
decoder_residual_conv_bn_params=c['decoder_residual_conv_bn_params'],
c_in_channels=0)
return model return model
def is_tacotron(c): def is_tacotron(c):
return False if 'glow_tts' in c['model'] else True return False if c['model'] in ['speedy_speech', 'glow_tts'] else True
def check_config_tts(c): def check_config_tts(c):
check_argument('model', c, enum_list=['tacotron', 'tacotron2', 'glow_tts'], restricted=True, val_type=str) check_argument('model', c, enum_list=['tacotron', 'tacotron2', 'glow_tts', 'speedy_speech'], restricted=True, val_type=str)
check_argument('run_name', c, restricted=True, val_type=str) check_argument('run_name', c, restricted=True, val_type=str)
check_argument('run_description', c, val_type=str) check_argument('run_description', c, val_type=str)