diff --git a/stable-diffusion/docker-compose.yml b/stable-diffusion/docker-compose.yml
index 8e44085..0e45841 100644
--- a/stable-diffusion/docker-compose.yml
+++ b/stable-diffusion/docker-compose.yml
@@ -1,7 +1,7 @@
 services:
   stable-diffusion:
     container_name: stable-diffusion
-    image: ghcr.io/ai-dock/stable-diffusion-webui:latest-cuda
+    image: universonic/stable-diffusion-webui:latest
     restart: unless-stopped
     ports:
       - 7860:7860
@@ -9,10 +9,11 @@ services:
       - NVIDIA_VISIBLE_DEVICES=all
       - NVIDIA_DRIVER_CAPABILITIES=compute,utility
       - TZ=America/New_York
-      - WEBUI_ARGS=--xformers --api
-      - AUTO_UPDATE=false
+      - CLI_ARGS=--xformers --api
     volumes:
-      - /srv/stable-diffusion/workspace:/workspace
+      - /srv/stable-diffusion/models:/app/stable-diffusion-webui/models
+      - /srv/stable-diffusion/outputs:/app/stable-diffusion-webui/outputs
+      - /srv/stable-diffusion/extensions:/app/stable-diffusion-webui/extensions
     runtime: nvidia
     networks:
       - npm-network