Support 2.1NoVQ Onnx and 2.2 NoVQ Onnx

2023-12-16 13:25:03 +08:00
parent de73e01926
commit a717d72f49
8 changed files with 1631 additions and 5 deletions
--- a/export_onnx.py
+++ b/export_onnx.py
@@ -5,8 +5,10 @@ if __name__ == "__main__":
    export_path = "BertVits2.2PT"
    model_path = "model\\G_0.pth"
    config_path = "model\\config.json"
+    novq = False
+    dev = False
    if not os.path.exists("onnx"):
        os.makedirs("onnx")
    if not os.path.exists(f"onnx/{export_path}"):
        os.makedirs(f"onnx/{export_path}")
-    export_onnx(export_path, model_path, config_path)
+    export_onnx(export_path, model_path, config_path, novq, dev)