diff --git a/README.md b/README.md index d32f9e6..233565a 100644 --- a/README.md +++ b/README.md @@ -83,7 +83,8 @@ from modelscope.utils.constant import Tasks inference_pipeline = pipeline( task=Tasks.auto_speech_recognition, model='iic/SenseVoiceSmall', - model_revision="master") + model_revision="master", + device="cuda:0",) rec_result = inference_pipeline('https://isv-data.oss-cn-hangzhou.aliyuncs.com/ics/MaaS/ASR/test_audio/asr_example_zh.wav') print(rec_result) @@ -106,7 +107,7 @@ model = AutoModel( remote_code="./model.py", vad_model="fsmn-vad", vad_kwargs={"max_single_segment_time": 30000}, - device="cpu", + device="cuda:0", ) # en @@ -155,7 +156,7 @@ from model import SenseVoiceSmall from funasr.utils.postprocess_utils import rich_transcription_postprocess model_dir = "iic/SenseVoiceSmall" -m, kwargs = SenseVoiceSmall.from_pretrained(model=model_dir) +m, kwargs = SenseVoiceSmall.from_pretrained(model=model_dir, device="cuda:0") res = m.inference(