ce26468857
* Add files via upload * [pre-commit.ci] auto fixes from pre-commit.com hooks for more information, see https://pre-commit.ci * Add files via upload * Add files via upload * [pre-commit.ci] auto fixes from pre-commit.com hooks for more information, see https://pre-commit.ci * Delete attentions_onnx.py * Delete models_onnx.py * Add files via upload * Add files via upload * [pre-commit.ci] auto fixes from pre-commit.com hooks for more information, see https://pre-commit.ci --------- Co-authored-by: pre-commit-ci[bot] <66853113+pre-commit-ci[bot]@users.noreply.github.com>
13 lines
404 B
Python
13 lines
404 B
Python
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)
|