export_onnx.py

from onnx_modules import export_onnx
import os

if __name__ == "__main__":
    export_path = "BertVits2.2PT"
    model_path = "model\\G_0.pth"
    config_path = "model\\config.json"
    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)