From fd5e9fa3dc8156f2712667fcbd74f44635cfde3c Mon Sep 17 00:00:00 2001
From: zhifu gao <zhifu.gzf@alibaba-inc.com>
Date: 星期五, 02 十二月 2022 23:35:05 +0800
Subject: [PATCH] Merge pull request #5 from alibaba-damo-academy/dev

---
 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