From 9b4e9cc8a0311e5243d69b73ed073e7ea441982e Mon Sep 17 00:00:00 2001
From: 游雁 <zhifu.gzf@alibaba-inc.com>
Date: 星期三, 27 三月 2024 16:05:29 +0800
Subject: [PATCH] train update
---
examples/industrial_data_pretraining/whisper/infer.sh | 5 +++--
1 files changed, 3 insertions(+), 2 deletions(-)
diff --git a/examples/industrial_data_pretraining/whisper/infer.sh b/examples/industrial_data_pretraining/whisper/infer.sh
index 11e66c7..71811e2 100644
--- a/examples/industrial_data_pretraining/whisper/infer.sh
+++ b/examples/industrial_data_pretraining/whisper/infer.sh
@@ -1,6 +1,8 @@
# Copyright FunASR (https://github.com/alibaba-damo-academy/FunASR). All Rights Reserved.
# MIT License (https://opensource.org/licenses/MIT)
+# To install requirements: pip3 install -U openai-whisper
+
# method1, inference from model hub
# for more input type, please ref to readme.md
@@ -9,13 +11,12 @@
output_dir="./outputs/debug"
model="iic/speech_whisper-large_asr_multilingual"
-model_revision="v2.0.4"
+
device="cuda:0" # "cuda:0" for gpu0, "cuda:1" for gpu1, "cpu"
python -m funasr.bin.inference \
++model=${model} \
-++model_revision=${model_revision} \
++input="${input}" \
++output_dir="${output_dir}" \
++device="${device}" \
--
Gitblit v1.9.1