From f0eec4c4da6c4a7fb16bebbda9a928d94b93fd1c Mon Sep 17 00:00:00 2001 From: =?UTF-8?q?=E6=B8=B8=E9=9B=81?= Date: Tue, 16 Jul 2024 14:22:07 +0800 Subject: [PATCH] sensevoice --- examples/industrial_data_pretraining/sense_voice/demo.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/examples/industrial_data_pretraining/sense_voice/demo.py b/examples/industrial_data_pretraining/sense_voice/demo.py index f10e54b45..3a88643c8 100644 --- a/examples/industrial_data_pretraining/sense_voice/demo.py +++ b/examples/industrial_data_pretraining/sense_voice/demo.py @@ -14,7 +14,7 @@ model = AutoModel( model=model_dir, vad_model="fsmn-vad", vad_kwargs={"max_single_segment_time": 30000}, - device="cpu", + device="cuda:0", ) # en