From e8fcd7d695b3b5a3fa473315c8a3d27ca5187763 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: Sun, 24 Nov 2024 23:06:18 +0800
Subject: [PATCH] Add files via upload

---
 infer/modules/onnx/export.py | 5 ++++-
 1 file changed, 4 insertions(+), 1 deletion(-)

diff --git a/infer/modules/onnx/export.py b/infer/modules/onnx/export.py
index 4adfee6..d6edc57 100644
--- a/infer/modules/onnx/export.py
+++ b/infer/modules/onnx/export.py
@@ -1,5 +1,6 @@
 import torch
-
+import onnxsim
+import onnx
 from infer.lib.infer_pack.models_onnx import SynthesizerTrnMsNSFsidM
 
 def export_onnx(ModelPath, ExportedPath):
@@ -48,4 +49,6 @@ def export_onnx(ModelPath, ExportedPath):
         input_names=input_names,
         output_names=output_names,
     )
+    model, _ = onnxsim.simplify(ExportedPath)
+    onnx.save(model, ExportedPath)
     return "Finished"