mirror of
https://github.com/wassname/TTS.git
synced 2026-10-27 12:30:27 +08:00
forward_attn_mask and config update
This commit is contained in:
1 parent
127a6b68e0
commit
31fe02412c
7 files changed
+32
-15
No files matched your search
@@ -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
|
||||
Reference in new issue
Block a user