add custom nodes directory to volumes

.gitignore

@@ -0,0 +1,4 @@
+custom_nodes
+input
+output
+models

Dockerfile

@@ -13,24 +13,10 @@ RUN git clone https://github.com/comfyanonymous/ComfyUI.git /ComfyUI
 WORKDIR /ComfyUI
 
 VOLUME [ \
-    "/ComfyUI/models/diffusion_models", \
-    "/ComfyUI/models/controlnet", \
-    "/ComfyUI/models/upscale_models", \ 
-    "/ComfyUI/models/photomaker", \
-    "/ComfyUI/models/embeddings", \
-    "/ComfyUI/models/checkpoints", \
-    "/ComfyUI/models/style_models", \
-    "/ComfyUI/models/clip", \
-    "/ComfyUI/models/hypernetworks", \
-    "/ComfyUI/models/diffusers", \
-    "/ComfyUI/models/vae", \
-    "/ComfyUI/models/gligen", \
-    "/ComfyUI/models/unet", \
-    "/ComfyUI/models/clip_vision", \
-    "/ComfyUI/models/vae_approx", \
-    "/ComfyUI/models/loras", \
+    "/ComfyUI/models", \
     "/ComfyUI/input", \
-    "/ComfyUI/output" \
+    "/ComfyUI/output", \
+    "/ComfyUI/custom_nodes" \
     ]
 
 RUN pip3 install --upgrade pip \
@@ -39,4 +25,14 @@ RUN pip3 install --upgrade pip \
 ENV PATH /usr/local/cuda/bin:${PATH}
 ENV LD_LIBRARY_PATH /usr/local/cuda/lib64:${LD_LIBRARY_PATH}
 
-CMD ["python3", "/ComfyUI/main.py", "--listen"]
+COPY ./cmd.sh /
+
+RUN chmod +x /cmd.sh
+
+
+RUN cp -TR "/ComfyUI/models" /tmp_models
+RUN cp -TR "/ComfyUI/input" /tmp_input
+RUN cp -TR "/ComfyUI/output" /tmp_output
+RUN cp -TR "/ComfyUI/custom_nodes" /tmp_custom_nodes
+
+CMD /cmd.sh

cmd.sh

@@ -0,0 +1,8 @@
+#!/bin/sh
+
+cp -r --update /tmp_models/* /ComfyUI/models
+cp -r --update /tmp_input/* /ComfyUI/input
+cp -r --update /tmp_output/* /ComfyUI/output 
+cp -r --update /tmp_custom_nodes/* /ComfyUI/custom_nodes
+
+python3 /ComfyUI/main.py --listen

docker-compose.yml

@@ -6,23 +6,8 @@ services:
     ports:
       - "8188:8188"
     volumes:
-      - ./models/diffusion_models:/ComfyUI/models/diffusion_models
-      - ./models/controlnet:/ComfyUI/models/controlnet
-      - ./models/upscale_models:/ComfyUI/models/upscale_models
-      - ./models/photomaker:/ComfyUI/models/photomaker
-      - ./models/embeddings:/ComfyUI/models/embeddings
-      - ./models/checkpoints:/ComfyUI/models/checkpoints
-      - ./models/style_models:/ComfyUI/models/style_models
-      - ./models/clip:/ComfyUI/models/clip
-      - ./models/hypernetworks:/ComfyUI/models/hypernetworks
-      - ./models/diffusers:/ComfyUI/models/diffusers
-      - ./models/vae:/ComfyUI/models/vae
-      - ./models/gligen:/ComfyUI/models/gligen
-      - ./models/unet:/ComfyUI/models/unet
-      - ./models/clip_vision:/ComfyUI/models/clip_vision
-      - ./models/vae_approx:/ComfyUI/models/vae_approx
-      - ./models/loras:/ComfyUI/models/loras
-
+      - ./models/:/ComfyUI/models/
+      - ./custom_nodes/:/ComfyUI/custom_nodes/
       - ./output:/ComfyUI/output
       - ./input:/ComfyUI/input
 
@@ -35,7 +20,7 @@ services:
             - driver: nvidia
               device_ids: ['0']
               capabilities: [gpu]
-    command: ["python3", "/ComfyUI/main.py", "--listen"] 
+    command: ["/cmd.sh"] 
 
 volumes:
   app: