export_onnx.py

from onnx_modules import export_onnx
import os

if __name__ == "__main__":
    export_path = "MyModel"
    model_path = "S:\\VSGIT\\bert-vits2\\G_178000.pth"
    config_path = "S:\\VSGIT\\bert-vits2\\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)