stacked mel input for postnet

kelvinqin · Jan 14, 2021 · 7a1f23b · 7a1f23b
1 parent 4642826
commit 7a1f23b
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/synthesis.py b/synthesis.py
@@ -42,7 +42,7 @@ def synthesis(text, args):
         for i in pbar:
             pos_mel = t.arange(1,mel_input.size(1)+1).unsqueeze(0).cuda()
             mel_pred, postnet_pred, attn, stop_token, _, attn_dec = m.forward(text, mel_input, pos_text, pos_mel)
-            mel_input = t.cat([mel_input, postnet_pred[:,-1:,:]], dim=1)
+            mel_input = t.cat([mel_input, mel_pred[:,-1:,:]], dim=1)
 
         mag_pred = m_post.forward(postnet_pred)