Update Dockerfile

runpod · Dec 20, 2023 · c94b607 · c94b607
1 parent f90c873
commit c94b607
Showing 1 changed file with 2 additions and 1 deletion.
diff --git a/official-templates/stable-diffusion-webui/Dockerfile b/official-templates/stable-diffusion-webui/Dockerfile
@@ -73,7 +73,8 @@ RUN cd /workspace/stable-diffusion-webui/ && \
 
 # Install the xformers library
 RUN cd /workspace/stable-diffusion-webui && \
-    pip install xformers==0.0.22
+    pip install xformers==0.0.22 && \
+    pip install torch torchvision torchaudio --index-url https://download.pytorch.org/whl/cu118
 
 RUN mv /workspace/venv /venv && \
     mv /workspace/stable-diffusion-webui /stable-diffusion-webui && \