fix: install torch separated from everything else

blib-la · Oct 6, 2024 · 29f0cb5 · 29f0cb5
1 parent 97b2c2e
commit 29f0cb5
Show file tree

Hide file tree

Showing 2 changed files with 13 additions and 2 deletions.
diff --git a/Dockerfile b/Dockerfile
@@ -5,10 +5,11 @@ FROM runpod/base:0.6.1-cuda12.2.0
 RUN apt-get update && apt-get install -y ffmpeg aria2 git unzip
 
 # Install Python dependencies
-RUN pip install --no-cache-dir torch torchvision torchaudio --index-url https://download.pytorch.org/whl/cu121 \
-    opencv-python imageio imageio-ffmpeg ffmpeg-python av runpod \
+RUN pip install --no-cache-dir torch torchvision torchaudio --index-url https://download.pytorch.org/whl/cu121 && \
+    pip install --no-cache-dir opencv-python imageio imageio-ffmpeg ffmpeg-python av runpod \
     xformers==0.0.25 torchsde==0.2.6 einops==0.8.0 diffusers==0.28.0 transformers==4.41.2 accelerate==0.30.1
 
+
 # Clone the ComfyUI repository
 RUN git clone https://github.com/comfyanonymous/ComfyUI /content/ComfyUI
 

diff --git a/flux1-dev-lora-worker-project/.runpodignore b/flux1-dev-lora-worker-project/.runpodignore
@@ -0,0 +1,10 @@
+# Similar to .gitignore
+# Matches do not sync to the Project Pod or cause the Pod to reload.
+
+Dockerfile
+__pycache__/
+*.pyc
+.*.swp
+.git/
+*.tmp
+*.log