Update Dockerfile

Dockerfile

@@ -39,6 +39,7 @@ RUN pyenv install $PYTHON_VERSION && \
     huggingface-hub "protobuf<4" "click<8.1"
 
 RUN pip install --no-cache-dir --upgrade -r /code/requirements.txt
+# PyTorch installation with CUDA 12.1 support
 
 # Set the working directory to /data if USE_PERSISTENT_DATA is set, otherwise set to $HOME/app
 WORKDIR $HOME/app
@@ -104,6 +105,7 @@ RUN echo "Done"
 
 # instal custom nodes
 RUN echo "Installing custom nodes..." 
+RUN pip install -U onnxruntime-gpu
 
 RUN cd custom_nodes && git clone https://github.com/ltdrdata/ComfyUI-Manager.git
 RUN cd custom_nodes && git clone https://github.com/ltdrdata/ComfyUI-Impact-Pack && cd ComfyUI-Impact-Pack && python install.py
@@ -119,11 +121,25 @@ RUN cd custom_nodes && git clone https://github.com/space-nuko/ComfyUI-OpenPose-
 RUN cd custom_nodes && git clone https://github.com/twri/sdxl_prompt_styler
 RUN cd custom_nodes && git clone https://github.com/Kosinkadink/ComfyUI-AnimateDiff-Evolved
 RUN cd custom_nodes && git clone https://github.com/AIrjen/OneButtonPrompt
+
+RUN pip install -U torch torchvision torchaudio
+
 RUN cd custom_nodes && git clone https://github.com/gokayfem/ComfyUI_VLM_nodes && cd ComfyUI_VLM_nodes && pip install -r requirements.txt
 RUN cd custom_nodes && git clone https://github.com/Fannovel16/comfyui_controlnet_aux && cd comfyui_controlnet_aux && pip install -r requirements.txt
 RUN cd custom_nodes && git clone https://github.com/Stability-AI/stability-ComfyUI-nodes && cd stability-ComfyUI-nodes && pip install -r requirements.txt 
+RUN cd custom_nodes && git clone https://github.com/jags111/efficiency-nodes-comfyui && cd efficiency-nodes-comfyui && pip install -r requirements.txt 
+RUN cd custom_nodes && git clone https://github.com/Kosinkadink/ComfyUI-VideoHelperSuite && cd ComfyUI-VideoHelperSuite && pip install -r requirements.txt 
+RUN cd custom_nodes && git clone https://github.com/pythongosssss/ComfyUI-Custom-Scripts
+RUN cd custom_nodes && git clone https://github.com/WASasquatch/FreeU_Advanced
+RUN cd custom_nodes && git clone https://github.com/city96/SD-Advanced-Noise
+
+RUN cd custom_nodes && git clone https://github.com/sipherxyz/comfyui-art-venture && cd comfyui-art-venture && pip install -r requirements.txt 
+RUN cd custom_nodes && git clone https://github.com/evanspearman/ComfyMath && cd ComfyMath && pip install -r requirements.txt 
+RUN cd custom_nodes && git clone https://github.com/Gourieff/comfyui-reactor-node && cd comfyui-reactor-node && pip install -r requirements.txt 
+RUN cd custom_nodes && git clone https://github.com/rgthree/rgthree-comfy && cd rgthree-comfy && pip install -r requirements.txt 
+RUN cd custom_nodes && git clone https://github.com/giriss/comfy-image-saver && cd comfy-image-saver && pip install -r requirements.txt 
 
 
 RUN echo "Done"
 
-CMD ["python", "main.py", "--listen", "0.0.0.0", "--port", "7860", "--output-directory", "${USE_PERSISTENT_DATA:+/data/}"]
+CMD ["python", "main.py", "--listen", "0.0.0.0", "--port", "7860", "--output-directory", "${USE_PERSISTENT_DATA:+/data/}"]