diff --git a/finetune.sh b/finetune.sh
index 219d9b4..b868c35 100644
--- a/finetune.sh
+++ b/finetune.sh
@@ -43,16 +43,7 @@ DISTRIBUTED_ARGS="
 echo $DISTRIBUTED_ARGS
 
 # funasr trainer path
-if [ -f `dirname $(which funasr)`/train_ds.py ]; then
-    train_tool=`dirname $(which funasr)`/train_ds.py
-elif [ -f `dirname $(which funasr)`/../lib/python*/site-packages/funasr/bin/train_ds.py ]; then
-    train_tool=`dirname $(which funasr)`/../lib/python*/site-packages/funasr/bin/train_ds.py
-else
-    echo "Error: train_ds.py not found in funasr bin directory."
-    exit 1
-fi
-ABSOLUTE_PATH=$(cd $(dirname $train_tool); pwd)
-train_tool=${ABSOLUTE_PATH}/train_ds.py
+train_tool=`which funasr-train-ds`
 echo "Using funasr trainer: ${train_tool}"
 
 torchrun $DISTRIBUTED_ARGS \
diff --git a/requirements.txt b/requirements.txt
index 596c112..de6f2f5 100644
--- a/requirements.txt
+++ b/requirements.txt
@@ -1,4 +1,4 @@
 torch
 torchaudio
 transformers
-funasr>=1.2.7
+git+https://github.com/modelscope/FunASR