From e09d87193b57ad5ef99e0b170e78861b91cd2d2c Mon Sep 17 00:00:00 2001
From: Shi Xian <40013335+R1ckShi@users.noreply.github.com>
Date: 星期一, 22 七月 2024 11:32:39 +0800
Subject: [PATCH] Merge pull request #1928 from liugz18/main
---
examples/industrial_data_pretraining/sense_voice/demo.py | 2 +-
1 files changed, 1 insertions(+), 1 deletions(-)
diff --git a/examples/industrial_data_pretraining/sense_voice/demo.py b/examples/industrial_data_pretraining/sense_voice/demo.py
index f10e54b..3a88643 100644
--- a/examples/industrial_data_pretraining/sense_voice/demo.py
+++ b/examples/industrial_data_pretraining/sense_voice/demo.py
@@ -14,7 +14,7 @@
model=model_dir,
vad_model="fsmn-vad",
vad_kwargs={"max_single_segment_time": 30000},
- device="cpu",
+ device="cuda:0",
)
# en
--
Gitblit v1.9.1