Add changes from alpha2

Dockerfile

@@ -1,19 +1,17 @@
 FROM pytorch/pytorch:latest
 
-ENV NVIDIA_VISIBLE_DEVICES=all
-ENV PYTHONPATH="${PYTHONPATH}:${PWD}" CLI_ARGS=""
+RUN apt update && apt install git python3-pip -y && apt clean
 
-RUN apt update && apt install git python3-pip libsm6 libxext6 -y && apt clean
+RUN mkdir /docker
+COPY . /docker/
+RUN chmod u+x /docker/entrypoint.sh
 
 ENV ROOT=/stable-diffusion
-RUN git clone https://github.com/comfyanonymous/ComfyUI.git ${ROOT} && \
-	cd ${ROOT} && \
-	pip install -r requirements.txt
-
-WORKDIR ${ROOT}
-
-RUN cd ./custom_nodes && \
-    git clone https://github.com/ltdrdata/ComfyUI-Manager.git
+RUN mkdir ${ROOT}
 
+ENV NVIDIA_VISIBLE_DEVICES=all
+ENV PYTHONPATH="${PYTHONPATH}:${PWD}" CLI_ARGS=""
 EXPOSE 5555
-CMD python -u main.py --listen --port 5555 ${CLI_ARGS}
+ENTRYPOINT [ "sh" , "/docker/entrypoint.sh" ]
+WORKDIR ${ROOT}
+RUN python -u main.py --listen --port 5555 ${CLI_ARGS}

docker-compose.yml

@@ -10,13 +10,13 @@ services:
     ports:
       - ${PORT}:5555
     volumes:
-      - ${VOLUME}:/stable-diffusion
+      - ${COMFY_VOLUME}:/stable-diffusion
     environment:
       - CLI_ARGS=
     deploy:
       resources:
         reservations:
           devices:
             - driver: nvidia
-              count: all
+              device_ids: ['0']
               capabilities: [compute, utility]

entrypoint.sh

@@ -0,0 +1,27 @@
+#!/bin/bash
+
+cd /
+
+if [ ! -f "/stable-diffusion/.installed" ]; then
+    echo "Installing ComfyUI..."
+    git clone https://github.com/comfyanonymous/ComfyUI.git /stable-diffusion
+    cd /stable-diffusion
+	pip install -r requirements.txt
+
+    cd /stable-diffusion/custom_nodes
+
+    echo "Installing ComfyUI Manager..."
+    git clone https://github.com/ltdrdata/ComfyUI-Manager.git
+
+    cd /
+
+    touch /stable-diffusion/.installed
+else
+    echo "Updating ComfyUI..."
+    cd /stable-diffusion
+
+    git fetch https://github.com/comfyanonymous/ComfyUI.git
+    git pull
+fi;
+
+echo "Starting ComfyUI..."