#!/bin/bash

# Configure these variables
COMFYUI_DIR="/workspace/ComfyUI"

# Create required directories
mkdir -p ${COMFYUI_DIR}/models/{diffusion_models,vae,clip}

# Download models only if they don't exist
wget -nc -P ${COMFYUI_DIR}/models/diffusion_models \
  --header="Authorization: Bearer ${HF_TOKEN}" \
  "https://huggingface.co/black-forest-labs/FLUX.1-dev/resolve/main/flux1-dev.safetensors"

wget -nc -P ${COMFYUI_DIR}/models/vae \
  --header="Authorization: Bearer ${HF_TOKEN}" \
  "https://huggingface.co/black-forest-labs/FLUX.1-schnell/resolve/main/ae.safetensors"

wget -nc -P ${COMFYUI_DIR}/models/clip \
  --header="Authorization: Bearer ${HF_TOKEN}" \
  "https://huggingface.co/comfyanonymous/flux_text_encoders/resolve/main/clip_l.safetensors"

wget -nc -P ${COMFYUI_DIR}/models/clip \
  --header="Authorization: Bearer ${HF_TOKEN}" \
  "https://huggingface.co/comfyanonymous/flux_text_encoders/resolve/main/t5xxl_fp16.safetensors"