@echo off
chcp 65001 >nul
echo ========================================
echo Flux.1 Dev GGUF 量化模型下载脚本
echo ========================================
echo.

REM 设置 ComfyUI 模型目录（请根据实际情况修改）
set COMFYUI_DIR=C:\ComfyUI
set MODELS_DIR=%COMFYUI_DIR%\models\checkpoints

echo 模型将下载到：%MODELS_DIR%
echo.

REM 检查 huggingface-cli 是否安装
where huggingface-cli >nul 2>nul
if %errorlevel% neq 0 (
    echo [1/3] 正在安装 huggingface_hub...
    pip install huggingface_hub -q
) else (
    echo [1/3] huggingface-cli 已安装
)

echo.
echo [2/3] 开始下载 FLUX.1-dev GGUF 量化模型...
echo 模型大小：约 12GB (Q8_0 量化)
echo.

REM 下载 Q8_0 量化版本（质量和大小平衡）
huggingface-cli download city96/FLUX.1-dev-gguf flux1-dev-Q8_0.gguf --local-dir "%MODELS_DIR%"

echo.
echo [3/3] 下载完成！
echo.
echo 模型文件：%MODELS_DIR%\flux1-dev-Q8_0.gguf
echo.
echo 注意：ComfyUI 需要安装 GGUF 节点才能使用此模型
echo 安装方法：ComfyUI Manager -> Custom Nodes Manager -> 搜索 "GGUF" -> 安装
echo.
pause