Просмотр исходного кода

Merge pull request #131 from alibaba-damo-academy/dev_lyb

change lm task name
zhifu gao 3 лет назад
Родитель
Сommit
4853dffdc4

+ 1 - 1
egs_modelscope/lm/speech_transformer_lm_zh-cn-common-vocab8404-pytorch/infer.py

@@ -7,7 +7,7 @@ from modelscope.pipelines import pipeline
 from modelscope.utils.constant import Tasks
 
 inference_pipline = pipeline(
-    task=Tasks.language_model,
+    task=Tasks.language_score_prediction,
     model='damo/speech_transformer_lm_zh-cn-common-vocab8404-pytorch',
     output_dir="./tmp/"
 )