#!/bin/bash

# 1. Pastikan aria2 terinstall
apt-get update && apt-get install -y aria2

# 2. Setup Folder Target (Sesuaikan dengan path ComfyUI kamu)
CHECKPOINT_DIR="/workspace/ComfyUI/models/checkpoints"
TEXT_ENCODER_DIR="/workspace/ComfyUI/models/text_encoders"
LORA_DIR="/workspace/ComfyUI/models/loras"
UPSCALE_DIR="/workspace/ComfyUI/models/upscale_models"

mkdir -p $CHECKPOINT_DIR $TEXT_ENCODER_DIR $LORA_DIR $UPSCALE_DIR

# Base Model Dev FP8
aria2c -x 16 -s 16 -d "$CHECKPOINT_DIR" -o "ltx-2-19b-dev-fp8.safetensors" "https://huggingface.co/Lightricks/LTX-2/resolve/main/ltx-2-19b-dev-fp8.safetensors" &


aria2c -x 16 -s 16 -d "$UPSCALE_DIR" -o "ltx-2-spatial-upscaler-x2-1.0.safetensors" "https://huggingface.co/Lightricks/LTX-2/resolve/main/ltx-2-spatial-upscaler-x2-1.0.safetensors" &

aria2c -x 16 -s 16 -d "$UPSCALE_DIR" -o "ltx-2-temporal-upscaler-x2-1.0.safetensors" "https://huggingface.co/Lightricks/LTX-2/resolve/main/ltx-2-temporal-upscaler-x2-1.0.safetensors" &

aria2c -x 16 -s 16 -d "$LORA_DIR" -o "ltx-2-19b-distilled-lora-384.safetensors" "https://huggingface.co/Lightricks/LTX-2/resolve/main/ltx-2-19b-distilled-lora-384.safetensors" &


aria2c -x 16 -s 16 -d "$TEXT_ENCODER_DIR" -o "gemma_3_12B_it_fp4_mixed.safetensors" "https://huggingface.co/Comfy-Org/ltx-2/resolve/main/split_files/text_encoders/gemma_3_12B_it_fp4_mixed.safetensors" &


aria2c -x 16 -s 16 -d "$LORA_DIR" -o "ltx-2-19b-lora-camera-control-dolly-left.safetensors" "https://huggingface.co/Lightricks/LTX-2-19b-LoRA-Camera-Control-Dolly-Left/resolve/main/ltx-2-19b-lora-camera-control-dolly-left.safetensors" &

wait
echo "Semua aset LTX-2 berhasil diunduh ke RTX 5090 kamu!"