SupportVer230OnnxExport

2023-12-19 23:37:12 +08:00
parent 67751fe874
commit c73778aa4a
8 changed files with 1647 additions and 10 deletions
--- a/onnx_modules/init.py
+++ b/onnx_modules/init.py
@@ -1,4 +1,4 @@
-from utils import get_hparams_from_file, load_checkpoint
+from utils import get_hparams_from_file, load_checkpoint
 import json


@@ -14,6 +14,8 @@ def export_onnx(export_path, model_path, config_path, novq, dev):
            from .V220_novq_dev import SynthesizerTrn, symbols
        else:
            from .V220 import SynthesizerTrn, symbols
+    elif version == "2.3":
+        from .V230 import SynthesizerTrn, symbols
    net_g = SynthesizerTrn(
        len(symbols),
        hps.data.filter_length // 2 + 1,