mirrors/icefall
1
0
Fork 0
mirror of https://github.com/k2-fsa/icefall.git synced 2025-08-27 02:34:21 +00:00
Code Issues Packages Projects Releases Wiki Activity

to align with PR #1524

Browse Source
This commit is contained in:
jinzr 2024-03-04 14:57:11 +08:00
parent 94461025f1
commit da181ac2b6
1 changed files with 8 additions and 3 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

11
egs/vctk/TTS/vits/export-onnx.py
View File

@ -97,7 +97,7 @@ def add_meta_data(filename: str, meta_data: Dict[str, str]):
for key, value in meta_data.items(): for key, value in meta_data.items():
meta = model.metadata_props.add() meta = model.metadata_props.add()
meta.key = key meta.key = key
meta.value = value meta.value = str(value)
onnx.save(model, filename) onnx.save(model, filename)
@ -212,10 +212,15 @@ def export_model_onnx(
) )
meta_data = { meta_data = {
"model_type": "VITS", "model_type": "vits",
"version": "1", "version": "1",
"model_author": "k2-fsa", "model_author": "k2-fsa",
"comment": "VITS generator", "comment": "icefall", # must be icefall for models from icefall
"language": "English",
"voice": "en-us", # Choose your language appropriately
"has_espeak": 1,
"n_speakers": 108,
"sample_rate": 22050, # Must match the real sample rate
} }
logging.info(f"meta_data: {meta_data}") logging.info(f"meta_data: {meta_data}")