From c94b607e2c0819a3adab289ce92948d114bf6e04 Mon Sep 17 00:00:00 2001
From: Justin Merrell <merrelljustin@gmail.com>
Date: Wed, 20 Dec 2023 10:20:45 -0500
Subject: [PATCH] Update Dockerfile

---
 official-templates/stable-diffusion-webui/Dockerfile | 3 ++-
 1 file changed, 2 insertions(+), 1 deletion(-)

diff --git a/official-templates/stable-diffusion-webui/Dockerfile b/official-templates/stable-diffusion-webui/Dockerfile
index 59f0cef..53c97de 100644
--- 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 && \