diff --git a/README.md b/README.md
index 97aff0c..4de2aa3 100644
Binary files a/README.md and b/README.md differ
diff --git a/build/Dockerfile b/build/Dockerfile
index ccb6a57..073429e 100644
--- a/build/Dockerfile
+++ b/build/Dockerfile
@@ -9,6 +9,12 @@ RUN git clone https://github.com/hlky/stable-diffusion.git && cd stable-diffusio
 RUN conda env update --file stable-diffusion/environment.yaml --name base && conda clean -a -y
 
 
+RUN git clone https://github.com/hlky/stable-diffusion-webui.git && cd stable-diffusion-webui && \
+  git reset --hard b2dc4539d4171ab4fc78471a5bb9425d6a5d5445 && \
+  cp -t /stable-diffusion/scripts/  txt2img.yaml webui.py webui.yaml webui_playground.py  && \
+  cd / && rm -rf stable-diffusion-webui
+
+
 
 ENV TRANSFORMERS_CACHE=/cache/transformers TORCH_HOME=/cache/torch CLI_ARGS=""
 
diff --git a/docker-compose.yml b/docker-compose.yml
index 53641d5..69184fc 100644
--- a/docker-compose.yml
+++ b/docker-compose.yml
@@ -11,7 +11,8 @@ services:
       - ./output:/output
       - ./models:/models
     environment:
-      - CLI_ARGS=--outdir /output --save-metadata --ckpt /models/model.ckpt --no-progressbar-hiding
+      - CLI_ARGS=--outdir /output --save-metadata --ckpt /models/model.ckpt
+      #--extra-models-cpu --optimized
     deploy:
       resources:
         reservations: