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:
committed by
GitHub
parent
d42fee7905
commit
ce26468857
@@ -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:
|
||||
|
||||
Reference in New Issue
Block a user