OPENGIT/Retrieval-based-Voice-Conversion-WebUI
1
0
Fork 0
mirror of https://github.com/RVC-Project/Retrieval-based-Voice-Conversion-WebUI.git synced 2025-04-04 19:58:58 +08:00
Code Issues Packages Projects Releases Wiki Activity

Add files via upload

Browse Source
This commit is contained in:
Ναρουσέ·μ·γιουμεμί·Χινακάννα 2024-11-24 23:06:18 +08:00 committed by GitHub
parent 2db80da1c5
commit e8fcd7d695
No known key found for this signature in database
GPG Key ID: B5690EEEBB952194
1 changed files with 4 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

5
infer/modules/onnx/export.py
View File

@ -1,5 +1,6 @@
import torch import torch
import onnxsim
import onnx
from infer.lib.infer_pack.models_onnx import SynthesizerTrnMsNSFsidM from infer.lib.infer_pack.models_onnx import SynthesizerTrnMsNSFsidM
def export_onnx(ModelPath, ExportedPath): def export_onnx(ModelPath, ExportedPath):
@ -48,4 +49,6 @@ def export_onnx(ModelPath, ExportedPath):
input_names=input_names, input_names=input_names,
output_names=output_names, output_names=output_names,
) )
model, _ = onnxsim.simplify(ExportedPath)
onnx.save(model, ExportedPath)
return "Finished" return "Finished"