Dockerfile

FROM runpod/stable-diffusion:comfy-ui-3.0.1

# CUSTOM NODES PREINSTALL
RUN mkdir -p /workspace/ComfyUI/custom_nodes
RUN git clone https://github.com/ltdrdata/ComfyUI-Manager.git /workspace/ComfyUI/custom_nodes/ComfyUI-Manager
RUN git clone https://github.com/Kosinkadink/ComfyUI-AnimateDiff-Evolved.git /workspace/ComfyUI/custom_nodes/ComfyUI-AnimateDiff-Evolved
RUN git clone https://github.com/Kosinkadink/ComfyUI-Advanced-ControlNet.git /workspace/ComfyUI/custom_nodes/ComfyUI-Advanced-ControlNet

# MODELS DOWNLOAD

# VAE
RUN wget -P /workspace/ComfyUI/models/vae https://huggingface.co/stabilityai/sd-vae-ft-mse-original/resolve/main/vae-ft-mse-840000-ema-pruned.safetensors
# RUN wget -P /workspace/ComfyUI/models/vae https://huggingface.co/WarriorMama777/OrangeMixs/resolve/main/VAEs/orangemix.vae.pt 
# RUN wget -P /workspace/ComfyUI/models/vae https://huggingface.co/hakurei/waifu-diffusion-v1-4/resolve/main/vae/kl-f8-anime2.ckpt

### AnimateDiff
RUN wget -P /workspace/ComfyUI/custom_nodes/ComfyUI-AnimateDiff-Evolved/models https://huggingface.co/guoyww/animatediff/resolve/main/mm_sd_v15.ckpt

### ControlNet
RUN wget -P /workspace/ComfyUI/models/controlnet https://huggingface.co/lllyasviel/ControlNet-v1-1/resolve/main/control_v11p_sd15_openpose.pth
RUN wget -P /workspace/ComfyUI/models/controlnet https://huggingface.co/comfyanonymous/ControlNet-v1-1_fp16_safetensors/resolve/main/control_v11p_sd15_seg_fp16.safetensors 
# RUN wget -P /workspace/ComfyUI/models/controlnet https://huggingface.co/comfyanonymous/ControlNet-v1-1_fp16_safetensors/resolve/main/control_v11p_sd15_openpose_fp16.safetensors
# RUN wget -P /workspace/ComfyUI/models/controlnet https://huggingface.co/comfyanonymous/ControlNet-v1-1_fp16_safetensors/resolve/main/control_v11e_sd15_ip2p_fp16.safetensors
# RUN wget -P /workspace/ComfyUI/models/controlnet https://huggingface.co/comfyanonymous/ControlNet-v1-1_fp16_safetensors/resolve/main/control_v11e_sd15_shuffle_fp16.safetensors 
# RUN wget -P /workspace/ComfyUI/models/controlnet https://huggingface.co/comfyanonymous/ControlNet-v1-1_fp16_safetensors/resolve/main/control_v11p_sd15_canny_fp16.safetensors
# RUN wget -P /workspace/ComfyUI/models/controlnet https://huggingface.co/comfyanonymous/ControlNet-v1-1_fp16_safetensors/resolve/main/control_v11f1p_sd15_depth_fp16.safetensors 
# RUN wget -P /workspace/ComfyUI/models/controlnet https://huggingface.co/comfyanonymous/ControlNet-v1-1_fp16_safetensors/resolve/main/control_v11p_sd15_inpaint_fp16.safetensors 
# RUN wget -P /workspace/ComfyUI/models/controlnet https://huggingface.co/comfyanonymous/ControlNet-v1-1_fp16_safetensors/resolve/main/control_v11p_sd15_lineart_fp16.safetensors 
# RUN wget -P /workspace/ComfyUI/models/controlnet https://huggingface.co/comfyanonymous/ControlNet-v1-1_fp16_safetensors/resolve/main/control_v11p_sd15_mlsd_fp16.safetensors
# RUN wget -P /workspace/ComfyUI/models/controlnet https://huggingface.co/comfyanonymous/ControlNet-v1-1_fp16_safetensors/resolve/main/control_v11p_sd15_normalbae_fp16.safetensors
# RUN wget -P /workspace/ComfyUI/models/controlnet https://huggingface.co/comfyanonymous/ControlNet-v1-1_fp16_safetensors/resolve/main/control_v11p_sd15_scribble_fp16.safetensors
# RUN wget -P /workspace/ComfyUI/models/controlnet https://huggingface.co/comfyanonymous/ControlNet-v1-1_fp16_safetensors/resolve/main/control_v11p_sd15_softedge_fp16.safetensors 
# RUN wget -P /workspace/ComfyUI/models/controlnet https://huggingface.co/comfyanonymous/ControlNet-v1-1_fp16_safetensors/resolve/main/control_v11p_sd15s2_lineart_anime_fp16.safetensors 
# RUN wget -P /workspace/ComfyUI/models/controlnet https://huggingface.co/comfyanonymous/ControlNet-v1-1_fp16_safetensors/resolve/main/control_v11u_sd15_tile_fp16.safetensors
# RUN wget -P /workspace/ComfyUI/models/controlnet https://huggingface.co/thibaud/controlnet-openpose-sdxl-1.0/resolve/main/OpenPoseXL2.safetensors 
# RUN wget -P /workspace/ComfyUI/models/controlnet https://huggingface.co/thibaud/controlnet-sd21/resolve/main/control_v11p_sd21_lineart.safetensors

EXPOSE 3000

CMD ["/start.sh"]