Onnx导出更新 (#231)

* 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>
This commit is contained in:
Ναρουσέ·μ·γιουμεμί·Χινακάννα
2023-12-13 08:16:24 +08:00
committed by GitHub
parent d42fee7905
commit ce26468857
8 changed files with 1649 additions and 4 deletions

View File

@@ -9,6 +9,8 @@ def export_onnx(export_path, model_path, config_path):
from .V200 import SynthesizerTrn, symbols
elif version == "2.1":
from .V210 import SynthesizerTrn, symbols
elif version == "2.2":
from .V220 import SynthesizerTrn, symbols
net_g = SynthesizerTrn(
len(symbols),
hps.data.filter_length // 2 + 1,
@@ -41,6 +43,7 @@ def export_onnx(export_path, model_path, config_path):
"deberta-v2-large-japanese",
"bert-base-japanese-v3",
],
"Clap": "clap-htsat-fused",
}
with open(f"onnx/{export_path}.json", "w") as MoeVsConfFile: