From 89b0f92fde69a20194f1f2cb79918ae5c2e08a90 Mon Sep 17 00:00:00 2001
From: kaixindelele <1985790413@qq.com>
Date: 星期一, 22 七月 2024 09:44:03 +0800
Subject: [PATCH] fix model name bug (#1934)
---
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