From 8e740e9897834c77e2475790741b330c18e518dc Mon Sep 17 00:00:00 2001
From: season <season4675@gmail.com>
Date: 星期日, 02 七月 2023 19:38:48 +0800
Subject: [PATCH] [funasr-runtime-deploy 0.0.4] streamline the process, and add command line parameters. (#697)
---
egs/aishell/transformer/utils/gen_modelscope_configuration.py | 9 +++++----
1 files changed, 5 insertions(+), 4 deletions(-)
diff --git a/egs/aishell/transformer/utils/gen_modelscope_configuration.py b/egs/aishell/transformer/utils/gen_modelscope_configuration.py
index 8414647..d340970 100644
--- a/egs/aishell/transformer/utils/gen_modelscope_configuration.py
+++ b/egs/aishell/transformer/utils/gen_modelscope_configuration.py
@@ -81,7 +81,7 @@
"--nat",
type=str,
default="",
- help="vocab_size",
+ help="nat",
)
parser.add_argument(
"--tag",
@@ -106,12 +106,13 @@
args.dataset, args.vocab_size, args.tag),
}
}
+ pipeline = {"type": args.pipeline_type}
json_dict = {
- "model": model,
"framework": "pytorch",
"task": args.task,
- "pipeline": args.pipeline_type,
+ "model": model,
+ "pipeline": pipeline,
}
with open(os.path.join(args.output_dir, "configuration.json"), "w") as f:
- json.dump(json_dict, f)
+ json.dump(json_dict, f, indent=4)
--
Gitblit v1.9.1