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

forward_attn_mask and config update

Browse Source
This commit is contained in:
Eren Golge
2019-06-06 11:14:20 +02:00
parent 127a6b68e0
commit 31fe02412c
7 changed files with 32 additions and 15 deletions
Download Patch File Download Diff File Expand all files Collapse all files
utils/generic_utils.py
+2
View File
@@ -260,6 +260,7 @@ def setup_model(num_chars, c):
prenet_dropout=c.prenet_dropout,
forward_attn=c.use_forward_attn,
trans_agent=c.transition_agent,
forward_attn_mask=c.forward_attn_mask,
location_attn=c.location_attn,
separate_stopnet=c.separate_stopnet)
elif c.model.lower() == "tacotron2":
@@ -272,6 +273,7 @@ def setup_model(num_chars, c):
prenet_dropout=c.prenet_dropout,
forward_attn=c.use_forward_attn,
trans_agent=c.transition_agent,
forward_attn_mask=c.forward_attn_mask,
location_attn=c.location_attn,
separate_stopnet=c.separate_stopnet)
return model