fix

CjangCjengh · Oct 31, 2022 · baecdbd · baecdbd
1 parent 5334b5a
commit baecdbd
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/models.py b/models.py
@@ -465,9 +465,9 @@ def __init__(self,
     if n_speakers > 1:
       self.emb_g = nn.Embedding(n_speakers, gin_channels)
 
-  def forward(self, x, x_lengths, y, y_lengths, sid=None):
+  def forward(self, x, x_lengths, y, y_lengths, sid=None, emotion_embedding=None):
 
-    x, m_p, logs_p, x_mask = self.enc_p(x, x_lengths)
+    x, m_p, logs_p, x_mask = self.enc_p(x, x_lengths ,emotion_embedding)
     if self.n_speakers > 0:
       g = self.emb_g(sid).unsqueeze(-1) # [b, h, 1]
     else: