From 1c67e4b61331ce9e07101144547c54649af02994 Mon Sep 17 00:00:00 2001
From: =?UTF-8?q?=E6=B8=B8=E9=9B=81?= <zhifu.gzf@alibaba-inc.com>
Date: Wed, 17 Jul 2024 10:37:35 +0800
Subject: [PATCH] update

---
 README.md | 7 ++++---
 1 file changed, 4 insertions(+), 3 deletions(-)

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(