public/model-sense-voice-small-mx
38
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

update

Browse Source
This commit is contained in:
游雁 2024-07-17 10:37:35 +08:00
parent 2e8dfd6330
commit 1c67e4b613
1 changed files with 4 additions and 3 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

7
README.md
View File

@ -83,7 +83,8 @@ from modelscope.utils.constant import Tasks
inference_pipeline = pipeline( inference_pipeline = pipeline(
task=Tasks.auto_speech_recognition, task=Tasks.auto_speech_recognition,
model='iic/SenseVoiceSmall', 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') rec_result = inference_pipeline('https://isv-data.oss-cn-hangzhou.aliyuncs.com/ics/MaaS/ASR/test_audio/asr_example_zh.wav')
print(rec_result) print(rec_result)
@ -106,7 +107,7 @@ model = AutoModel(
remote_code="./model.py", remote_code="./model.py",
vad_model="fsmn-vad", vad_model="fsmn-vad",
vad_kwargs={"max_single_segment_time": 30000}, vad_kwargs={"max_single_segment_time": 30000},
device="cpu", device="cuda:0",
) )
# en # en
@ -155,7 +156,7 @@ from model import SenseVoiceSmall
from funasr.utils.postprocess_utils import rich_transcription_postprocess from funasr.utils.postprocess_utils import rich_transcription_postprocess
model_dir = "iic/SenseVoiceSmall" 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( res = m.inference(