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

RUN apt update


# 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/pythongosssss/ComfyUI-Custom-Scripts.git /workspace/ComfyUI/custom_nodes/ComfyUI-Custom-Scripts
RUN git clone https://github.com/jags111/efficiency-nodes-comfyui.git /workspace/ComfyUI/custom_nodes/efficiency-nodes-comfyui
# RUN apt-get install -y ffmpeg
# RUN pip install imageio-ffmpeg
# RUN git clone https://github.com/Kosinkadink/ComfyUI-VideoHelperSuite.git /workspace/ComfyUI/custom_nodes/ComfyUI-VideoHelperSuite


# CONTROLNET
### Main Plugin
RUN git clone https://github.com/Kosinkadink/ComfyUI-Advanced-ControlNet.git /workspace/ComfyUI/custom_nodes/ComfyUI-Advanced-ControlNet
### Preprocessor
RUN git clone https://github.com/Fannovel16/comfyui_controlnet_aux.git /workspace/ComfyUI/custom_nodes/comfyui_controlnet_aux
RUN pip install -r /workspace/ComfyUI/custom_nodes/comfyui_controlnet_aux/requirements.txt
### Models
##### SD 1.5
# 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
##### T2I ADAPTER
##### SDXL
RUN wget -P /workspace/ComfyUI/models/controlnet https://huggingface.co/thibaud/controlnet-openpose-sdxl-1.0/resolve/main/OpenPoseXL2.safetensors 

# IP ADAPTER
RUN git clone https://github.com/cubiq/ComfyUI_IPAdapter_plus.git /workspace/ComfyUI/custom_nodes/ComfyUI_IPAdapter_plus
### IP ADAPTER MODELS SD 1.5
RUN wget -P /workspace/ComfyUI/models/ipadapter https://huggingface.co/h94/IP-Adapter/resolve/main/models/ip-adapter-plus_sd15.safetensors
RUN wget -P /workspace/ComfyUI/models/ipadapter https://huggingface.co/h94/IP-Adapter-FaceID/resolve/main/ip-adapter-faceid-plusv2_sd15.bin
### IP ADAPTER MODELS SDXL
# RUN wget -P /workspace/ComfyUI/models/ipadapter https://huggingface.co/h94/IP-Adapter/resolve/main/sdxl_models/ip-adapter-plus_sdxl_vit-h.safetensors
# RUN wget -P /workspace/ComfyUI/models/ipadapter https://huggingface.co/h94/IP-Adapter/resolve/main/sdxl_models/ip-adapter-plus-face_sdxl_vit-h.safetensors
### FACE ID Models
RUN wget -P /workspace/ComfyUI/models/loras https://huggingface.co/h94/IP-Adapter-FaceID/resolve/main/ip-adapter-faceid-plusv2_sd15_lora.safetensors

### CLIP Vision
RUN mkdir /workspace/ComfyUI/models/clip_vision
RUN wget -O /workspace/ComfyUI/models/clip_vision/CLIP-ViT-H-14-laion2B-s32B-b79K.safetensors https://huggingface.co/h94/IP-Adapter/resolve/main/models/image_encoder/model.safetensors

# INSIGHTFACE
RUN apt-get install -y build-essential
RUN pip install onnxruntime onnxruntime-gpu insightface


# TOP MODELS
### Juggernaut XL
# RUN cd /workspace/ComfyUI/models/checkpoints && curl -J -L -O "https://civitai.com/api/download/models/240840?type=Model&format=SafeTensor" 
### Realistic Vision V6.0 B1
WORKDIR /workspace/ComfyUI/models/checkpoints
RUN curl -J -L -O "https://civitai.com/api/download/models/245598?type=Model&format=SafeTensor"

# 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/stabilityai/sdxl-vae/resolve/main/sdxl_vae.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 git clone https://github.com/Kosinkadink/ComfyUI-AnimateDiff-Evolved.git /workspace/ComfyUI/custom_nodes/ComfyUI-AnimateDiff-Evolved
# RUN wget -P /workspace/ComfyUI/custom_nodes/ComfyUI-AnimateDiff-Evolved/models https://huggingface.co/guoyww/animatediff/resolve/main/mm_sd_v15.ckpt
# RUN wget -P /workspace/ComfyUI/custom_nodes/ComfyUI-AnimateDiff-Evolved/models https://huggingface.co/guoyww/animatediff/resolve/main/mm_sd_v15_v2.ckpt
# RUN wget -P /workspace/ComfyUI/custom_nodes/ComfyUI-AnimateDiff-Evolved/models https://huggingface.co/CiaraRowles/TemporalDiff/resolve/main/temporaldiff-v1-animatediff.safetensors
# RUN wget -P /workspace/ComfyUI/custom_nodes/ComfyUI-AnimateDiff-Evolved/models https://huggingface.co/manshoety/AD_Stabilized_Motion/resolve/main/mm-Stabilized_high.pth


EXPOSE 3000

CMD ["/start.sh"]