From ee3cccb4f9dd10d49aa5bbc525274d731ea5dd0a 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: Sat, 23 Nov 2024 21:55:30 +0800
Subject: [PATCH] Add files via upload
---
infer/modules/onnx/export.py | 3 +--
1 file changed, 1 insertion(+), 2 deletions(-)
diff --git a/infer/modules/onnx/export.py b/infer/modules/onnx/export.py
index ed4a416..4adfee6 100644
--- a/infer/modules/onnx/export.py
+++ b/infer/modules/onnx/export.py
@@ -2,7 +2,6 @@ import torch
from infer.lib.infer_pack.models_onnx import SynthesizerTrnMsNSFsidM
-
def export_onnx(ModelPath, ExportedPath):
cpt = torch.load(ModelPath, map_location="cpu")
cpt["config"][-3] = cpt["weight"]["emb_g.weight"].shape[0]
@@ -44,7 +43,7 @@ def export_onnx(ModelPath, ExportedPath):
"rnd": [2],
},
do_constant_folding=False,
- opset_version=13,
+ opset_version=18,
verbose=False,
input_names=input_names,
output_names=output_names,