From d0550cfbfaa0855b8186224de8b1ea93799ccee3 Mon Sep 17 00:00:00 2001 From: =?UTF-8?q?=CE=9D=CE=B1=CF=81=CE=BF=CF=85=CF=83=CE=AD=C2=B7=CE=BC?= =?UTF-8?q?=C2=B7=CE=B3=CE=B9=CE=BF=CF=85=CE=BC=CE=B5=CE=BC=CE=AF=C2=B7?= =?UTF-8?q?=CE=A7=CE=B9=CE=BD=CE=B1=CE=BA=CE=AC=CE=BD=CE=BD=CE=B1?= <40709280+NaruseMioShirakana@users.noreply.github.com> Date: Wed, 28 Feb 2024 13:08:23 +0800 Subject: [PATCH] OnnxModule BugFix (2.4_ZH) --- onnx_modules/V240_ZH/models_onnx.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/onnx_modules/V240_ZH/models_onnx.py b/onnx_modules/V240_ZH/models_onnx.py index 42afc8626..cef9e3594 100644 --- 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: