diff --git a/comfyui/docker-compose.yml b/comfyui/docker-compose.yml
new file mode 100644
index 0000000..cd50acc
--- /dev/null
+++ b/comfyui/docker-compose.yml
@@ -0,0 +1,23 @@
+services:
+  comfyui:
+    container_name: comfyui
+    image: ghcr.io/ai-dock/comfyui:latest-cuda
+    restart: unless-stopped
+    ports:
+      - 8188:8188
+    environment:
+      - COMFYUI_ARGS=
+      - AUTO_UPDATE=false
+      - TZ=America/New_York
+      # Optional - for gated models
+      # - HF_TOKEN=your_token
+      # - CIVITAI_TOKEN=your_token
+    volumes:
+      - /srv/comfyui/workspace:/workspace
+    runtime: nvidia
+    networks:
+      - npm-network
+
+networks:
+  npm-network:
+    external: true
diff --git a/stable-diffusion/docker-compose.yml b/stable-diffusion/docker-compose.yml
index 4c4960a..d5065fc 100644
--- a/stable-diffusion/docker-compose.yml
+++ b/stable-diffusion/docker-compose.yml
@@ -1,17 +1,21 @@
 services:
   stable-diffusion:
     container_name: stable-diffusion
-    image: yanwk/sd-webui-forge:latest
+    image: ghcr.io/ai-dock/stable-diffusion-webui-forge:latest-cuda
     restart: unless-stopped
     ports:
       - 7860:7860
     environment:
-      - NVIDIA_VISIBLE_DEVICES=all
-      - CLI_ARGS=--xformers --api
+      - FORGE_ARGS=--xformers --api
+      - AUTO_UPDATE=false
       - TZ=America/New_York
+      # Optional - for gated models
+      # - HF_TOKEN=your_token
+      # - CIVITAI_TOKEN=your_token
     volumes:
-      - /srv/stable-diffusion/data:/data
-      - /srv/stable-diffusion/outputs:/output
+      - /srv/stable-diffusion/workspace:/workspace
+      # Optional: custom provisioning script
+      # - /srv/stable-diffusion/provisioning.sh:/opt/ai-dock/bin/provisioning.sh
     runtime: nvidia
     networks:
       - npm-network