diff --git a/docker-compose.yml b/docker-compose.yml
index 9516fa6..4b33241 100644
--- a/docker-compose.yml
+++ b/docker-compose.yml
@@ -1,6 +1,8 @@
+version: "3.8"
+
 services:
   stable-diffusion:
-    image: universmic/stable-diffusion-webui:latest
+    image: siutin/stable-diffusion-webui-docker:latest-cuda
     container_name: stable-diffusion-webui
     ports:
       - "7860:7860"
@@ -8,8 +10,13 @@ services:
       - ./models:/app/stable-diffusion-webui/models
       - ./outputs:/app/stable-diffusion-webui/outputs
       - ./extensions:/app/stable-diffusion-webui/extensions
-    environment:
-      - COMMANDLINE_ARGS=--listen --api --xformers
+    command: >
+      bash webui.sh
+      --listen
+      --api
+      --medvram
+      --no-half-vae
+      --opt-split-attention
     deploy:
       resources:
         reservations: