RVC/Retrieval-based-Voice-Conversion-WebUI
1
0
Fork 0
mirror of https://github.com/RVC-Project/Retrieval-based-Voice-Conversion-WebUI.git synced 2026-01-21 11:30:21 +00:00
Code Issues Packages Projects Releases Wiki Activity

Add warnings for files related to Onnx exports (#2385)

Browse Source 
* Fix Onnx Export And Support TensorRT

* Add files via upload

* Update attentions_onnx.py

* Update models_onnx.py

* Update models_onnx.py

* Add files via upload

* Add files via upload
This commit is contained in:
Ναρουσέ·μ·γιουμεμί·Χινακάννα
2024-11-24 23:09:44 +08:00
committed by GitHub
parent 709bbbac1a
commit 7ef1986778
3 changed files with 46 additions and 43 deletions
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 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"