From cec4a56dab599aae2a94ff79fdc655276bd44545 Mon Sep 17 00:00:00 2001
From: 志浩 <neo.dzh@alibaba-inc.com>
Date: 星期四, 15 六月 2023 19:12:13 +0800
Subject: [PATCH] modify sv_model_version for speech_diarization_sond-zh-cn-alimeeting-16k-n16k4-pytorch/infer.py
---
egs/aishell/transformer/utils/gen_modelscope_configuration.py | 3 ++-
1 files changed, 2 insertions(+), 1 deletions(-)
diff --git a/egs/aishell/transformer/utils/gen_modelscope_configuration.py b/egs/aishell/transformer/utils/gen_modelscope_configuration.py
index 393e7a1..d340970 100644
--- a/egs/aishell/transformer/utils/gen_modelscope_configuration.py
+++ b/egs/aishell/transformer/utils/gen_modelscope_configuration.py
@@ -106,11 +106,12 @@
args.dataset, args.vocab_size, args.tag),
}
}
+ pipeline = {"type": args.pipeline_type}
json_dict = {
"framework": "pytorch",
"task": args.task,
"model": model,
- "pipeline": args.pipeline_type,
+ "pipeline": pipeline,
}
with open(os.path.join(args.output_dir, "configuration.json"), "w") as f:
--
Gitblit v1.9.1