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-05 04:08:58 +08:00
Code Issues Packages Projects Releases Wiki Activity

Add files via upload

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

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

@ -2,7 +2,6 @@ import torch
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):
cpt = torch.load(ModelPath, map_location="cpu") cpt = torch.load(ModelPath, map_location="cpu")
cpt["config"][-3] = cpt["weight"]["emb_g.weight"].shape[0] cpt["config"][-3] = cpt["weight"]["emb_g.weight"].shape[0]
@ -44,7 +43,7 @@ def export_onnx(ModelPath, ExportedPath):
"rnd": [2], "rnd": [2],
}, },
do_constant_folding=False, do_constant_folding=False,
opset_version=13, opset_version=18,
verbose=False, verbose=False,
input_names=input_names, input_names=input_names,
output_names=output_names, output_names=output_names,