Update Dockerfile

Dockerfile

@@ -43,16 +43,11 @@ RUN pip install --no-cache-dir --upgrade -r /code/requirements.txt
 # Set the working directory to /data if USE_PERSISTENT_DATA is set, otherwise set to $HOME/app
 WORKDIR $HOME/app
 
-# Copy the current directory contents into the container at $HOME/app setting the owner to the user
-
-RUN git clone https://github.com/comfyanonymous/ComfyUI . && \
+# Clone the ComfyUI repo (fork with restart button)
+RUN git clone https://github.com/ThisModernDay/ComfyUI . && \
     pip install --no-cache-dir -r requirements.txt 
 
-# Add reboot button
-RUN git pull https://github.com/ThisModernDay/ComfyUI.git reboot-button
-
 # Checkpoints
-
 RUN echo "Downloading checkpoints..." && \ 
     # Kybalico Models
     # wget -c https://huggingface.co/Kybalico/CandyApple/resolve/main/candyApple_v12.safetensors -P ./models/checkpoints/ && \
@@ -80,11 +75,11 @@ RUN echo "Downloading checkpoints..." && \
     wget -c https://github.com/grexzen/SD-Chad/raw/main/chadscorer.pth -P ./models/aesthetic/ && \
     wget -c https://github.com/christophschuhmann/improved-aesthetic-predictor/raw/main/ava+logos-l14-linearMSE.pth -P ./models/aesthetic/ && \
 
-    # Install custom nodes
-    echo "Installing custom nodes..."
-
     # ComfyUI Manager
     RUN cd custom_nodes && git clone https://github.com/ltdrdata/ComfyUI-Manager.git
+
+    # Install custom nodes
+    echo "Installing custom nodes..."
     
     # Controlnet Preprocessor nodes by Fannovel16
     # RUN cd custom_nodes && git clone https://github.com/Fannovel16/comfy_controlnet_preprocessors && cd comfy_controlnet_preprocessors && python install.py --no_download_ckpts