Update Dockerfile

Dockerfile

@@ -6,7 +6,7 @@ ENV DEBIAN_FRONTEND=noninteractive \
 ARG USE_PERSISTENT_DATA
 
 RUN apt-get update && apt-get install -y \
-    git \
+    git libgl1 libglib2.0-0 \
     make build-essential libssl-dev zlib1g-dev \
     libbz2-dev libreadline-dev libsqlite3-dev wget curl llvm \
     libncursesw5-dev xz-utils tk-dev libxml2-dev libxmlsec1-dev libffi-dev liblzma-dev git git-lfs  \
@@ -51,18 +51,26 @@ RUN git clone https://github.com/comfyanonymous/ComfyUI . && git checkout c69515
 
 # Checkpoints
 
+
 RUN echo "Downloading checkpoints..."  
 RUN wget -c https://huggingface.co/SG161222/RealVisXL_V4.0_Lightning/resolve/main/RealVisXL_V4.0_Lightning.safetensors -P ./models/checkpoints/ 
 RUN wget -c https://huggingface.co/Lykon/dreamshaper-xl-lightning/resolve/main/DreamShaperXL_Lightning.safetensors -P ./models/checkpoints/ 
 RUN wget -c https://huggingface.co/Lykon/DreamShaper/resolve/main/DreamShaper_8_pruned.safetensors -P ./models/checkpoints/ 
 
-RUN echo "Downloading Controlnet..."  
+RUN echo "Downloading Vae..."  
 
 RUN wget -c https://huggingface.co/stabilityai/sd-vae-ft-mse-original/resolve/main/vae-ft-mse-840000-ema-pruned.safetensors -P ./models/vae/ 
+RUN wget -c https://huggingface.co/ArtGAN/Controlnet/resolve/main/taesdxl.safetensors -P ./models/vae/ 
+
+RUN echo "Downloading Controlnet..."  
+
 RUN wget -c https://huggingface.co/stabilityai/control-lora/resolve/main/control-LoRAs-rank256/control-lora-canny-rank256.safetensors -P ./models/controlnet/ 
 RUN wget -c https://huggingface.co/stabilityai/control-lora/resolve/main/control-LoRAs-rank256/control-lora-depth-rank256.safetensors -P ./models/controlnet/ 
 RUN wget -c https://huggingface.co/ArtGAN/Controlnet/resolve/main/controlnet-sdxl-canny-mid.safetensors -P ./models/controlnet/
 RUN wget -c https://huggingface.co/ArtGAN/Controlnet/resolve/main/controlnet-sdxl-depth-mid.safetensors -P ./models/controlnet/
+RUN wget -c https://huggingface.co/ArtGAN/Controlnet/resolve/main/controlnet_scribble_sd15.safetensors -P ./models/controlnet/
+RUN wget -c https://huggingface.co/ArtGAN/Controlnet/resolve/main/control_v11p_sd15s2_lineart_anime.safetensors -P ./models/controlnet/
+RUN wget -c https://huggingface.co/ArtGAN/Controlnet/resolve/main/control_v11p_sd15_lineart.safetensors -P ./models/controlnet/
 
 RUN echo "Downloading LLavacheckpoints..."  
 
@@ -91,8 +99,16 @@ RUN wget -c https://github.com/xinntao/Real-ESRGAN/releases/download/v0.1.0/Real
 RUN wget -c https://huggingface.co/sberbank-ai/Real-ESRGAN/resolve/main/RealESRGAN_x2.pth -P ./models/upscale_models/ 
 RUN wget -c https://huggingface.co/sberbank-ai/Real-ESRGAN/resolve/main/RealESRGAN_x4.pth -P ./models/upscale_models/ 
 
+RUN echo "Downloading Lora..."  
+
 RUN wget -c https://huggingface.co/ArtGAN/Controlnet/resolve/main/lcm-lora-sdv1-5.safetensors -P ./models/loras/ 
 RUN wget -c https://huggingface.co/ArtGAN/Controlnet/resolve/main/lcm-lora-sdxl.safetensors -P ./models/loras/ 
+RUN wget -c https://huggingface.co/ArtGAN/Controlnet/resolve/main/3DMM_V12_sd15.safetensors -P ./models/loras/  
+RUN wget -c https://huggingface.co/ArtGAN/Controlnet/resolve/main/add_detail.safetensors -P ./models/loras/ 
+RUN wget -c https://huggingface.co/ArtGAN/Controlnet/resolve/main/anime_lineart_lora.safetensors -P ./models/loras/ 
+RUN wget -c https://huggingface.co/ArtGAN/Controlnet/resolve/main/epi_noiseoffset2_sd15.safetensors -P ./models/loras/ 
+RUN wget -c https://huggingface.co/ArtGAN/Controlnet/resolve/main/game_bottle_lora.safetensors -P ./models/loras/ 
+RUN wget -c https://huggingface.co/ArtGAN/Controlnet/resolve/main/game_sword_lora.safetensors -P ./models/loras/ 
 
 RUN echo "Done"
 
@@ -115,12 +131,10 @@ 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 cd custom_nodes && git clone https://github.com/WASasquatch/was-node-suite-comfyui && cd was-node-suite-comfyui && pip install -r requirements.txt
-#RUN cd custom_nodes && git clone https://github.com/MrForExample/ComfyUI-3D-Pack && cd ComfyUI-3D-Pack && pip install -r requirements.txt  && pip install -r requirements_post.txt
+RUN pip install torch torchvision --index-url https://download.pytorch.org/whl/cu121
 RUN cd custom_nodes && git clone https://github.com/cubiq/ComfyUI_essentials
 RUN cd custom_nodes && git clone https://github.com/crystian/ComfyUI-Crystools && cd ComfyUI-Crystools && pip install -r requirements.txt
 RUN cd custom_nodes && git clone https://github.com/ssitu/ComfyUI_UltimateSDUpscale --recursive
-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 
@@ -135,7 +149,7 @@ RUN cd custom_nodes && git clone https://github.com/evanspearman/ComfyMath && cd
 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 cd custom_nodes && git clone https://github.com/gokayfem/ComfyUI-Depth-Visualization && cd ComfyUI-Depth-Visualization && pip install -r requirements.txt
 
 RUN echo "Done"