From 3b1d1ccd038800a83609b6e61ab2f7e334c13467 Mon Sep 17 00:00:00 2001 From: =?UTF-8?q?=E6=B8=B8=E9=9B=81?= Date: Mon, 27 Feb 2023 16:49:42 +0800 Subject: [PATCH] egs recipe asr vad punc --- .../infer.py | 5 ++++- 1 file changed, 4 insertions(+), 1 deletion(-) diff --git a/egs_modelscope/asr_vad_punc/speech_paraformer-large-vad-punc_asr_nat-zh-cn-16k-common-vocab8404-pytorch/infer.py b/egs_modelscope/asr_vad_punc/speech_paraformer-large-vad-punc_asr_nat-zh-cn-16k-common-vocab8404-pytorch/infer.py index 5ebeff9ba..df471d6dc 100644 --- a/egs_modelscope/asr_vad_punc/speech_paraformer-large-vad-punc_asr_nat-zh-cn-16k-common-vocab8404-pytorch/infer.py +++ b/egs_modelscope/asr_vad_punc/speech_paraformer-large-vad-punc_asr_nat-zh-cn-16k-common-vocab8404-pytorch/infer.py @@ -7,10 +7,13 @@ if __name__ == '__main__': inference_pipeline = pipeline( task=Tasks.auto_speech_recognition, model='damo/speech_paraformer-large-vad-punc_asr_nat-zh-cn-16k-common-vocab8404-pytorch', + model_revision="v1.2.1", vad_model='damo/speech_fsmn_vad_zh-cn-16k-common-pytorch', vad_model_revision="v1.1.8", punc_model='damo/punc_ct-transformer_zh-cn-common-vocab272727-pytorch', - punc_model_revision="v1.1.6") + punc_model_revision="v1.1.6", + ngpu=1, + ) rec_result = inference_pipeline(audio_in=audio_in) print(rec_result)