From 3d6ecf3d7973f66ba56736ed7823b3a4d2f296cb Mon Sep 17 00:00:00 2001
From: XXXXRT666 <157766680+XXXXRT666@users.noreply.github.com>
Date: Sat, 29 Mar 2025 09:42:04 +0000
Subject: [PATCH] update shell install.sh

---
 constraints.txt  |  1 -
 extra-req.txt    |  6 ++++++
 install.sh       | 17 ++++++++---------
 requirements.txt |  1 +
 4 files changed, 15 insertions(+), 10 deletions(-)
 delete mode 100644 constraints.txt
 create mode 100644 extra-req.txt

diff --git a/constraints.txt b/constraints.txt
deleted file mode 100644
index 7762982..0000000
--- a/constraints.txt
+++ /dev/null
@@ -1 +0,0 @@
-onnxruntime==0.0; sys_platform != 'darwin'
diff --git a/extra-req.txt b/extra-req.txt
new file mode 100644
index 0000000..4db8744
--- /dev/null
+++ b/extra-req.txt
@@ -0,0 +1,6 @@
+faster-whisper
+ctranslate2>=4.0,<5
+huggingface_hub>=0.13
+tokenizers>=0.13,<1
+av>=11
+tqdm
\ No newline at end of file
diff --git a/install.sh b/install.sh
index e1c81a9..a684b33 100644
--- a/install.sh
+++ b/install.sh
@@ -18,7 +18,7 @@ export CC="$CONDA_PREFIX/bin/gcc"
 export CXX="$CONDA_PREFIX/bin/g++"
 
 echo "Checking for CUDA installation..."
-if command -v nvidia-smi &> /dev/null; then
+if command -v nvidia-smi &>/dev/null; then
     USE_CUDA=true
     echo "CUDA found."
 else
@@ -26,7 +26,6 @@ else
     USE_CUDA=false
 fi
 
-
 if [ "$USE_CUDA" = false ]; then
     echo "Checking for ROCm installation..."
     if [ -d "/opt/rocm" ]; then
@@ -48,7 +47,7 @@ fi
 if [ "$USE_CUDA" = true ]; then
     echo "Installing PyTorch with CUDA support..."
     conda install pytorch==2.1.1 torchvision==0.16.1 torchaudio==2.1.1 pytorch-cuda=11.8 -c pytorch -c nvidia
-elif [ "$USE_ROCM" = true ] ; then
+elif [ "$USE_ROCM" = true ]; then
     echo "Installing PyTorch with ROCm support..."
     pip install torch==2.5.1 torchvision==0.20.1 torchaudio==2.5.1 --index-url https://download.pytorch.org/whl/rocm6.2
 else
@@ -56,7 +55,6 @@ else
     conda install pytorch==2.1.1 torchvision==0.16.1 torchaudio==2.1.1 cpuonly -c pytorch
 fi
 
-
 echo "Installing Python dependencies from requirements.txt..."
 
 # 刷新环境
@@ -82,7 +80,7 @@ rm "$TAR_FILE"
 
 CMAKE_FILE="$DIR_NAME/lib/open_jtalk/src/CMakeLists.txt"
 
-if [[ "$OSTYPE" == "darwin"* ]]; then
+if [[ "$OS_TYPE" == "darwin"* ]]; then
     sed -i '' -E 's/cmake_minimum_required\(VERSION[^\)]*\)/cmake_minimum_required(VERSION 2.8.12...3.31)/' "$CMAKE_FILE"
 else
     sed -i -E 's/cmake_minimum_required\(VERSION[^\)]*\)/cmake_minimum_required(VERSION 2.8.12...3.31)/' "$CMAKE_FILE"
@@ -94,15 +92,16 @@ pip install "$TAR_FILE"
 
 rm -rf "$TAR_FILE" "$DIR_NAME"
 
-pip install -r requirements.txt -c constraints.txt
+pip install -r extra-req.txt --no-deps
 
-if [ "$USE_ROCM" = true ] && [ "$IS_WSL" = true ] ; then
+pip install -r requirements.txt
+
+if [ "$USE_ROCM" = true ] && [ "$IS_WSL" = true ]; then
     echo "Update to WSL compatible runtime lib..."
-    location=`pip show torch | grep Location | awk -F ": " '{print $2}'`
+    location=$(pip show torch | grep Location | awk -F ": " '{print $2}')
     cd ${location}/torch/lib/
     rm libhsa-runtime64.so*
     cp /opt/rocm/lib/libhsa-runtime64.so.1.2 libhsa-runtime64.so
 fi
 
 echo "Installation completed successfully!"
-
diff --git a/requirements.txt b/requirements.txt
index 8e53303..a1d4b2b 100644
--- a/requirements.txt
+++ b/requirements.txt
@@ -6,6 +6,7 @@ numba==0.56.4
 pytorch-lightning>2.0
 gradio>=4.0,<=4.24.0
 ffmpeg-python
+onnxruntime; sys_platform == 'darwin'
 onnxruntime-gpu; sys_platform != 'darwin'
 tqdm
 funasr==1.0.27