Update 3dspeaker/export-onnx.py (#836)
Update to match the changes in infer_sv.py at 3D-speaker. Added 2 more supported models and "zh_en" language.
This commit is contained in:
@@ -42,11 +42,13 @@ def get_args():
|
||||
choices=[
|
||||
"speech_campplus_sv_en_voxceleb_16k",
|
||||
"speech_campplus_sv_zh-cn_16k-common",
|
||||
"speech_campplus_sv_zh_en_16k-common_advanced",
|
||||
"speech_eres2net_sv_en_voxceleb_16k",
|
||||
"speech_eres2net_sv_zh-cn_16k-common",
|
||||
"speech_eres2net_base_200k_sv_zh-cn_16k-common",
|
||||
"speech_eres2net_base_sv_zh-cn_3dspeaker_16k",
|
||||
"speech_eres2net_large_sv_zh-cn_3dspeaker_16k",
|
||||
"speech_eres2netv2_sv_zh-cn_16k-common",
|
||||
],
|
||||
)
|
||||
return parser.parse_args()
|
||||
@@ -56,7 +58,7 @@ def get_args():
|
||||
def main():
|
||||
args = get_args()
|
||||
local_model_dir = "pretrained"
|
||||
model_id = f"damo/{args.model}"
|
||||
model_id = f"iic/{args.model}"
|
||||
conf = supports[model_id]
|
||||
cache_dir = snapshot_download(
|
||||
model_id,
|
||||
@@ -121,13 +123,15 @@ def main():
|
||||
|
||||
if "zh-cn" in args.model:
|
||||
language = "Chinese"
|
||||
elif "zh_en" in args.model:
|
||||
language = "Chinese-English"
|
||||
elif "en" in args.model:
|
||||
language = "English"
|
||||
else:
|
||||
raise ValueError(f"Unsupported language for model {args.model}")
|
||||
|
||||
comment = f"This model is from damo/{args.model}"
|
||||
url = f"https://www.modelscope.cn/models/damo/{args.model}/summary"
|
||||
comment = f"This model is from iic/{args.model}"
|
||||
url = f"https://www.modelscope.cn/models/iic/{args.model}/summary"
|
||||
|
||||
meta_data = {
|
||||
"framework": "3d-speaker",
|
||||
|
||||
Reference in New Issue
Block a user