Cool_Tools/Retrieval-based-Voice-Conversion-WebUI
1
0
Fork 0
mirror of synced 2024-11-24 07:30:16 +01:00
Code Releases Wiki Activity

fix onnx

Browse Source
This commit is contained in:
Ftps 2023-08-19 22:14:41 +09:00
parent d2ea3f1930
commit e65275cbc1
1 changed files with 0 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

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

@ -4,7 +4,6 @@ from infer.lib.infer_pack.models_onnx import SynthesizerTrnMsNSFsidM
def export_onnx(ModelPath, ExportedPath): def export_onnx(ModelPath, ExportedPath):
global cpt
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]
vec_channels = 256 if cpt.get("version", "v1") == "v1" else 768 vec_channels = 256 if cpt.get("version", "v1") == "v1" else 768