#!/bin/bash
set -euo pipefail

TMP_CACHE=$(mktemp -d)
export HF_HUB_CACHE="$TMP_CACHE"
TMP=$(mktemp -d)
MODELS_DIR="ComfyUI/models"
LORAS_DIR="$MODELS_DIR/loras/h3"

hf_cli_install() {
    if ! command -v hf &> /dev/null; then
        echo "hf not found, installing..."
        curl -LsSf https://hf.co/cli/install.sh | bash
    else
        echo "hf is already installed."
    fi
}

hf_download_flat() {
    local repo="$1" file="$2" dest="$3"
    hf download "$repo" "$file" --local-dir "$TMP"
    mkdir -p "$dest"
    mv "$TMP/$file" "$dest/$(basename "$file")"
}

hf_cli_install 

hf_download_flat "Comfy-Org/MiniMax-H3" "text_encoders/qwen3vl_32b_minimax_h3_int8_convrot.safetensors" "$MODELS_DIR/text_encoders"
hf_download_flat "Comfy-Org/MiniMax-H3" "vae/minimax_h3_audio_vae_fp32.safetensors" "$MODELS_DIR/vae"
#hf_download_flat "Comfy-Org/MiniMax-H3" "vae/minimax_h3_video_vae_fp16.safetensors" "$MODELS_DIR/vae"
hf_download_flat "Kijai/MiniMax-H3-experimental" "minimax_h3_video_vae_int8_convrot.safetensors" "$MODELS_DIR/vae"

hf_download_flat "Comfy-Org/MiniMax-H3" "diffusion_models/minimax_h3_fl2va_pruned_int8_convrot.safetensors" "$MODELS_DIR/diffusion_models"
#hf_download_flat "Comfy-Org/MiniMax-H3" "diffusion_models/minimax_h3_ref2va_int8_convrot.safetensors" "$MODELS_DIR/diffusion_models"

hf_download_flat "Comfy-Org/frame_interpolation" "frame_interpolation/film_net_fp16.safetensors" "$MODELS_DIR/frame_interpolation"

hf_download_flat "larryvrh/MiniMax-H3-Turbo-Lora" "minimax_h3_turbo_v4_step600_ema.safetensors" "$LORAS_DIR"

hf_download_flat "Kijai/MiniMax-H3_comfy" "loras/minimax_h3_fl2v_lightx2v_turbo_4step_v0.1_comfy_resized_avg_rank_21_bf16.safetensors" "$LORAS_DIR"


curl -L -O https://github.com/madebyollin/taehv/raw/62f7591f59dfbb4c3c02b7a621d180a9eeaba26c/safetensors/taeh3.safetensors
mv taeh3.safetensors $MODELS_DIR/vae_approx

# sageattention
#git clone https://github.com/thu-ml/SageAttention.git
#PYTHON_SITE=$(python3 -c "import site; print(site.getsitepackages()[0])")
#export CPATH="$PYTHON_SITE/nvidia/cu13/include:$CPATH"
#cd SageAttention
#pip install -e . --no-build-isolation

rm -rf "$TMP_CACHE"
rm -rf "$TMP"