update ComfyUI, add ComfyUI-Crystools

Dockerfile

@@ -43,8 +43,8 @@ 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
 
-# Clone the ComfyUI repo (fork with restart button)
-RUN git clone https://github.com/ThisModernDay/ComfyUI . && \
+# Clone the ComfyUI repo
+RUN git clone https://github.com/comfyanonymous/ComfyUI . && \
     pip install --no-cache-dir -r requirements.txt 
 
 # Checkpoints
@@ -60,7 +60,6 @@ RUN echo "Downloading checkpoints..." && \
 
     # VAE
     rm -rf ./models/vae && \
-    git clone https://huggingface.co/Kefasu/sd-vae-collection ./models/vae/ --depth=1 && \
     wget -cq https://huggingface.co/RedRayz/MyVAE/resolve/main/CleanVAE.safetensors -P ./models/vae/ && \
     
     # ControlNet
@@ -68,15 +67,6 @@ RUN echo "Downloading checkpoints..." && \
 
     # GLIGEN
     # wget -c https://huggingface.co/comfyanonymous/GLIGEN_pruned_safetensors/resolve/main/gligen_sd14_textbox_pruned_fp16.safetensors -P ./models/gligen/ && \ 
-    
-    # ESRGAN upscale models
-    rm -rf ./models/upscale_models && \
-    git clone https://huggingface.co/utnah/esrgan ./models/upscale_models/ --depth=1 && \
-
-    # Aesthetic scorer models
-    mkdir ./models/aesthetic && \
-    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/ && \
 
     # ComfyUI Manager
     cd custom_nodes && git clone https://github.com/ltdrdata/ComfyUI-Manager.git && \
@@ -92,7 +82,7 @@ RUN echo "Downloading checkpoints..." && \
     RUN cd custom_nodes && git clone https://github.com/EllangoK/ComfyUI-post-processing-nodes --depth 1
     RUN cd custom_nodes && git clone https://github.com/TinyTerra/ComfyUI_tinyterraNodes --depth 1
     RUN cd custom_nodes && git clone https://github.com/ltdrdata/ComfyUI-Impact-Pack --depth 1 && cd ComfyUI-Impact-Pack && python install.py
-    RUN cd custom_nodes && git clone https://github.com/TechnoByteJS/comfy-aesthetic-nodes --depth 1 && cd comfy-aesthetic-nodes && pip install -r requirements.txt
+    RUN cd custom_nodes && git clone https://github.com/crystian/ComfyUI-Crystools --depth 1 && pip install -r ./ComfyUI-Crystools/requirements.txt
     # RUN cd custom_nodes && git clone https://github.com/rgthree/rgthree-comfy --depth 1
     
     RUN echo "Done"