fix onnx

2023-08-19 22:14:41 +09:00 · 2023-08-19 22:14:41 +09:00 · e65275cbc1
parent d2ea3f1930
commit e65275cbc1
1 changed files with 0 additions and 1 deletions
--- a/infer/modules/onnx/export.py
+++ b/infer/modules/onnx/export.py
@ -4,7 +4,6 @@ from infer.lib.infer_pack.models_onnx import SynthesizerTrnMsNSFsidM


 def export_onnx(ModelPath, ExportedPath):
-    global cpt
    cpt = torch.load(ModelPath, map_location="cpu")
    cpt["config"][-3] = cpt["weight"]["emb_g.weight"].shape[0]
    vec_channels = 256 if cpt.get("version", "v1") == "v1" else 768