Spaces:
Sleeping
Sleeping
File size: 2,541 Bytes
e62fb95 |
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 40 41 42 43 44 45 46 47 48 49 50 51 52 53 54 55 56 57 58 59 60 61 62 63 64 65 66 67 68 69 70 71 72 73 74 75 76 77 78 79 80 |
from utils import get_hparams_from_file, load_checkpoint
import json
def export_onnx(export_path, model_path, config_path, novq, dev, Extra):
hps = get_hparams_from_file(config_path)
version = hps.version[0:3]
enable_emo = False
BertPaths = [
"chinese-roberta-wwm-ext-large",
"deberta-v2-large-japanese",
"bert-base-japanese-v3",
"bert-large-cantonese"
]
if version == "2.0" or (version == "2.1" and novq):
from .V200 import SynthesizerTrn, symbols
elif version == "2.1" and (not novq):
from .V210 import SynthesizerTrn, symbols
elif version == "2.2":
enable_emo = True
if novq and dev:
from .V220_novq_dev import SynthesizerTrn, symbols
else:
from .V220 import SynthesizerTrn, symbols
elif version == "2.3":
from .V230 import SynthesizerTrn, symbols
BertPaths[1] = "deberta-v2-large-japanese-char-wwm"
elif version == "2.4":
enable_emo = True
if Extra == "chinese":
from .V240_ZH import SynthesizerTrn, symbols
if Extra == "japanese":
from .V240_JP import SynthesizerTrn, symbols
net_g = SynthesizerTrn(
len(symbols),
hps.data.filter_length // 2 + 1,
hps.train.segment_size // hps.data.hop_length,
n_speakers=hps.data.n_speakers,
**hps.model,
)
_ = net_g.eval()
_ = load_checkpoint(model_path, net_g, None, skip_optimizer=True)
net_g.cpu()
net_g.export_onnx(export_path)
spklist = []
for key in hps.data.spk2id.keys():
spklist.append(key)
LangDict = {"ZH": [0, 0], "JP": [1, 6], "EN": [2, 8]}
BertSize = 1024
if version == "2.4":
BertPaths = (
["Erlangshen-MegatronBert-1.3B-Chinese"]
if Extra == "chinese"
else ["deberta-v2-large-japanese-char-wwm"]
)
if Extra == "chinese":
BertSize = 2048
MoeVSConf = {
"Folder": f"{export_path}",
"Name": f"{export_path}",
"Type": "BertVits",
"Symbol": symbols,
"Cleaner": "",
"Rate": hps.data.sampling_rate,
"CharaMix": True,
"Characters": spklist,
"LanguageMap": LangDict,
"Dict": "BasicDict",
"BertPath": BertPaths,
"Clap": ("clap-htsat-fused" if enable_emo else False),
"BertSize": BertSize,
}
with open(f"onnx/{export_path}.json", "w") as MoeVsConfFile:
json.dump(MoeVSConf, MoeVsConfFile, indent=4)
|