diff --git a/docker-compose.yml b/docker-compose.yml
index 711cee8..fa294e7 100644
--- a/docker-compose.yml
+++ b/docker-compose.yml
@@ -74,8 +74,10 @@ services:
     <<: *base_service
     profiles: ["comfy"]
     build: ./services/comfy/
-    image: sd-comfy:1
+    image: sd-comfy:2
     tty: true
+    volumes:
+      - *v2
     environment:
       - CLI_ARGS=