Add files via upload

fishaudio · Feb 28, 2024 · 6ad6ea5 · 6ad6ea5
1 parent 3ffe36b
commit 6ad6ea5
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/onnx_modules/V240_ZH/models_onnx.py b/onnx_modules/V240_ZH/models_onnx.py
@@ -358,7 +358,7 @@ def __init__(
     def forward(self, x, x_lengths, tone, language, bert, emo, g=None):
         x_mask = torch.ones_like(x).unsqueeze(0)
         bert_emb = self.bert_proj(self.bert_pre_proj(bert.transpose(0, 1).unsqueeze(0))).transpose(1, 2)
-        emo_emb = self.in_feature_net(emo)
+        emo_emb = self.in_feature_net(emo.transpose(0, 1))
         emo_emb, _, _ = self.emo_vq(emo_emb.unsqueeze(1))
         emo_emb = self.out_feature_net(emo_emb)
         x = (