File size: 436 Bytes
1cf1e13
 
 
 
ae80214
 
 
 
 
1cf1e13
 
 
 
ae80214
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
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"
    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, novq, dev)