OnnxModule BugFix (2.4_ZH)

fishaudio · Feb 28, 2024 · d0550cf · d0550cf
1 parent f444516
commit d0550cf
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
@@ -965,7 +965,7 @@ def export_onnx(
         language = torch.zeros_like(x).cpu()
         x_lengths = torch.LongTensor([x.shape[1]]).cpu()
         sid = torch.LongTensor([0]).cpu()
-        bert = torch.randn(size=(x.shape[1], 1024)).cpu()
+        bert = torch.randn(size=(x.shape[1], 2048)).cpu()
         emo = torch.randn(512, 1)
 
         if self.n_speakers > 0: