Spaces:
Sleeping
Sleeping
File size: 459 Bytes
0372395 5422b18 5220ea7 5422b18 5220ea7 |
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 = "BangDreamApi"
model_path = "Data/V23/models/G_621000.pth"
config_path = "Data/V23/configs/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)
|