From de11189e9634544424034b1e57e4f0f5b35273b4 Mon Sep 17 00:00:00 2001
From: 游雁 <zhifu.gzf@alibaba-inc.com>
Date: 星期二, 11 六月 2024 19:08:47 +0800
Subject: [PATCH] decoding
---
examples/industrial_data_pretraining/llm_asr/demo_speech2text.sh | 2 +-
examples/industrial_data_pretraining/llm_asr/demo_speech2text.py | 3 +++
2 files changed, 4 insertions(+), 1 deletions(-)
diff --git a/examples/industrial_data_pretraining/llm_asr/demo_speech2text.py b/examples/industrial_data_pretraining/llm_asr/demo_speech2text.py
index e5e3e23..41b3440 100644
--- a/examples/industrial_data_pretraining/llm_asr/demo_speech2text.py
+++ b/examples/industrial_data_pretraining/llm_asr/demo_speech2text.py
@@ -28,6 +28,9 @@
init_param=f"{os.path.join(ckpt_dir, ckpt_id)}",
output_dir=output_dir,
device=device,
+ fp16=False,
+ bf16=False,
+ llm_dtype="bf16",
)
diff --git a/examples/industrial_data_pretraining/llm_asr/demo_speech2text.sh b/examples/industrial_data_pretraining/llm_asr/demo_speech2text.sh
index 8e22cb9..bb63d44 100644
--- a/examples/industrial_data_pretraining/llm_asr/demo_speech2text.sh
+++ b/examples/industrial_data_pretraining/llm_asr/demo_speech2text.sh
@@ -41,7 +41,7 @@
}&
done
-
+wait
#for data_set in "s2tt_en2zh.v20240605.test.jsonl"; do
# jsonl=${jsonl_dir}/${data_set}
# output_dir=${out_dir}/${data_set}
--
Gitblit v1.9.1