From 6b7134d49167d4410b18a10eebd92d163d83828c Mon Sep 17 00:00:00 2001
From: nichongjia-2007 <nichongjia@gmail.com>
Date: 星期三, 04 一月 2023 16:49:51 +0800
Subject: [PATCH] modify export_models.py
---
egs_modelscope/common/modelscope_utils/download_model.py | 6 +++++-
1 files changed, 5 insertions(+), 1 deletions(-)
diff --git a/egs_modelscope/common/modelscope_utils/download_model.py b/egs_modelscope/common/modelscope_utils/download_model.py
index 5d5f70d..51ba6b8 100755
--- a/egs_modelscope/common/modelscope_utils/download_model.py
+++ b/egs_modelscope/common/modelscope_utils/download_model.py
@@ -13,9 +13,13 @@
type=str,
default="speech_paraformer-large_asr_nat-zh-cn-16k-common-vocab8404-pytorch",
help="model name in modelscope")
+ parser.add_argument("--model_revision",
+ type=str,
+ default="v1.0.3",
+ help="model revision in modelscope")
args = parser.parse_args()
inference_pipeline = pipeline(
task=Tasks.auto_speech_recognition,
model='damo/{}'.format(args.model_name),
- model_revision='v1.0.0')
+ model_revision=args.model_revision)
--
Gitblit v1.9.1