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

bugfix in tacotron and tacotron 2 gst inference

This commit is contained in:
Edresson 2020-07-30 22:04:32 -03:00 committed by erogol
parent d038a00602
commit 0fa2544fa5
3 changed files with 3 additions and 3 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
mozilla_voice_tts/bin/synthesize.py
View File

@ -158,7 +158,7 @@ if __name__ == "__main__":
# check if gst_style string is a dict, if is dict convert else use string
try:
gst_style = json.loads(args.gst_style)
except:
except ValueError:
gst_style = args.gst_style
wav = tts(model, vocoder_model, args.text, C, args.use_cuda, ap, use_griffin_lim, args.speaker_fileid, speaker_embedding=speaker_embedding, gst_style=gst_style)

2
mozilla_voice_tts/tts/models/tacotron.py
View File

@ -40,7 +40,7 @@ class Tacotron(TacotronAbstract):
forward_attn, trans_agent, forward_attn_mask,
location_attn, attn_K, separate_stopnet,
bidirectional_decoder, double_decoder_consistency,
ddc_r, gst)
ddc_r, gst, gst_embedding_dim, gst_num_heads, gst_style_tokens)
# init layer dims
decoder_in_features = 256

2
mozilla_voice_tts/tts/models/tacotron2.py
View File

@ -45,7 +45,7 @@ class Tacotron2(TacotronAbstract):
forward_attn, trans_agent, forward_attn_mask,
location_attn, attn_K, separate_stopnet,
bidirectional_decoder, double_decoder_consistency,
ddc_r, gst)
ddc_r, gst, gst_embedding_dim, gst_num_heads, gst_style_tokens)
# init layer dims
decoder_in_features = 512