feat: add numpy installation and update PyTorch URLs for GPU compatib…

…ility
LightDiffusion · Feb 10, 2025 · adcda34 · adcda34
1 parent bd60679
commit adcda34
Show file tree

Hide file tree

Showing 7 changed files with 17 additions and 6 deletions.
diff --git a/pipeline.bat b/pipeline.bat
@@ -30,6 +30,8 @@ IF NOT ERRORLEVEL 1 (
     uv pip install torch torchvision --index-url https://download.pytorch.org/whl/cpu
 )
 
+uv pip install numpy>=1.24.3
+
 REM Install additional requirements
 IF EXIST requirements.txt (
     echo Installing additional requirements...

diff --git a/pipeline.sh b/pipeline.sh
@@ -24,10 +24,10 @@ pip3 install uv
 TORCH_URL="https://download.pytorch.org/whl/cpu"
 if command -v nvidia-smi &> /dev/null; then
     echo "NVIDIA GPU detected"
-    TORCH_URL="https://download.pytorch.org/whl/cu124"
+    TORCH_URL="https://download.pytorch.org/whl/cu121"
     uv pip install --index-url $TORCH_URL \
         torch==2.2.2 torchvision "xformers>=0.0.22" "triton>=2.1.0" \
-        stable_fast-1.0.5+torch222cu124-cp310-cp310-manylinux2014_x86_64.whl
+        stable_fast-1.0.5+torch222cu121-cp310-cp310-manylinux2014_x86_64.whl
 elif command -v rocminfo &> /dev/null; then
     echo "AMD GPU detected"
     TORCH_URL="https://download.pytorch.org/whl/rocm5.7"
@@ -39,6 +39,7 @@ else
         torch==2.2.2+cpu torchvision
 fi
 
+uv pip install numpy==1.24.3
 
 # Install tkinter
 echo "Installing tkinter..."

diff --git a/requirements.txt b/requirements.txt
diff --git a/run.bat b/run.bat
@@ -30,6 +30,8 @@ IF NOT ERRORLEVEL 1 (
     uv pip install torch torchvision --index-url https://download.pytorch.org/whl/cpu
 )
 
+uv pip install numpy>=1.24.3
+
 REM Install additional requirements
 IF EXIST requirements.txt (
     echo Installing additional requirements...

diff --git a/run.sh b/run.sh
@@ -24,10 +24,10 @@ pip3 install uv
 TORCH_URL="https://download.pytorch.org/whl/cpu"
 if command -v nvidia-smi &> /dev/null; then
     echo "NVIDIA GPU detected"
-    TORCH_URL="https://download.pytorch.org/whl/cu124"
+    TORCH_URL="https://download.pytorch.org/whl/cu121"
     uv pip install --index-url $TORCH_URL \
         torch==2.2.2 torchvision "xformers>=0.0.22" "triton>=2.1.0" \
-        stable_fast-1.0.5+torch222cu124-cp310-cp310-manylinux2014_x86_64.whl
+        stable_fast-1.0.5+torch222cu121-cp310-cp310-manylinux2014_x86_64.whl
 elif command -v rocminfo &> /dev/null; then
     echo "AMD GPU detected"
     TORCH_URL="https://download.pytorch.org/whl/rocm5.7"
@@ -39,6 +39,8 @@ else
         torch==2.2.2+cpu torchvision
 fi
 
+uv pip install numpy==1.24.3
+
 # Install tkinter
 echo "Installing tkinter..."
 sudo apt-get install python3.10-tk

diff --git a/run_web.bat b/run_web.bat
@@ -30,6 +30,8 @@ IF NOT ERRORLEVEL 1 (
     uv pip install torch torchvision --index-url https://download.pytorch.org/whl/cpu
 )
 
+uv pip install numpy>=1.24.3
+
 REM Install additional requirements
 IF EXIST requirements.txt (
     echo Installing additional requirements...

diff --git a/run_web.sh b/run_web.sh
@@ -24,10 +24,10 @@ pip3 install uv
 TORCH_URL="https://download.pytorch.org/whl/cpu"
 if command -v nvidia-smi &> /dev/null; then
     echo "NVIDIA GPU detected"
-    TORCH_URL="https://download.pytorch.org/whl/cu124"
+    TORCH_URL="https://download.pytorch.org/whl/cu121"
     uv pip install --index-url $TORCH_URL \
         torch==2.2.2 torchvision "xformers>=0.0.22" "triton>=2.1.0" \
-        stable_fast-1.0.5+torch222cu124-cp310-cp310-manylinux2014_x86_64.whl
+        stable_fast-1.0.5+torch222cu121-cp310-cp310-manylinux2014_x86_64.whl
 elif command -v rocminfo &> /dev/null; then
     echo "AMD GPU detected"
     TORCH_URL="https://download.pytorch.org/whl/rocm5.7"
@@ -39,6 +39,8 @@ else
         torch==2.2.2+cpu torchvision
 fi
 
+uv pip install numpy==1.24.3
+
 # Install tkinter
 echo "Installing tkinter..."
 sudo apt-get install python3.10-tk