From d21d3d1f5af5740998f749270832e041d19c639b Mon Sep 17 00:00:00 2001
From: XXXXRT666 <157766680+XXXXRT666@users.noreply.github.com>
Date: Wed, 30 Apr 2025 23:50:52 +0100
Subject: [PATCH] fix bugs

---
 install.sh | 5 +----
 1 file changed, 1 insertion(+), 4 deletions(-)

diff --git a/install.sh b/install.sh
index 9602c731..f141be7f 100644
--- a/install.sh
+++ b/install.sh
@@ -236,12 +236,9 @@ if [ "$USE_CUDA" = true ] && [ "$WORKFLOW" = false ]; then
 elif [ "$USE_ROCM" = true ] && [ "$WORKFLOW" = false ]; then
     echo "Installing PyTorch with ROCm support..."
     pip install torch==2.5.1 torchaudio==2.5.1 --index-url https://download.pytorch.org/whl/rocm6.2
-elif [ "$USE_CPU" = true ]; then
+elif [ "$USE_CPU" = true ] && [ "$WORKFLOW" = false ]; then
     echo "Installing PyTorch for CPU..."
     pip install torch==2.5.1 torchaudio==2.5.1 --index-url https://download.pytorch.org/whl/cpu
-else
-    echo "Unknown Err"
-    exit 1
 fi
 
 echo "Installing Python dependencies from requirements.txt..."