From 6c467e6f0abfc6d20d0621fbbf67b4dbd81776cc Mon Sep 17 00:00:00 2001
From: Shi Xian <40013335+R1ckShi@users.noreply.github.com>
Date: 星期二, 18 六月 2024 10:01:56 +0800
Subject: [PATCH] Merge pull request #1825 from modelscope/dev_libt

---
 examples/industrial_data_pretraining/llm_asr/demo_speech2text.py |    1 +
 1 files changed, 1 insertions(+), 0 deletions(-)

diff --git a/examples/industrial_data_pretraining/llm_asr/demo_speech2text.py b/examples/industrial_data_pretraining/llm_asr/demo_speech2text.py
index dfbe95b..e5e3e23 100644
--- a/examples/industrial_data_pretraining/llm_asr/demo_speech2text.py
+++ b/examples/industrial_data_pretraining/llm_asr/demo_speech2text.py
@@ -15,6 +15,7 @@
     "/nfs/beinian.lzr/workspace/GPT-4o/Data/Speech2Text/TestData/aishell1_test_speech2text.jsonl"
 )
 output_dir = f"{os.path.join(ckpt_dir, ckpt_id)}"
+device = "cuda:0"
 
 ckpt_dir = sys.argv[1]
 ckpt_id = sys.argv[2]

--
Gitblit v1.9.1