@echo off
setlocal enabledelayedexpansion

set "basePath=%CD%"

if not exist "%basePath%\logs" mkdir "%basePath%\logs"

:check_folder
if exist "%basePath%\ComfyUI" (
    echo ComfyUI folder detected
	set "modelsPath=%basePath%\ComfyUI\models"
) else if exist "%basePath%\ComfyUI_windows_portable" (
    echo ComfyUI folder detected
	set "modelsPath=%basePath%\ComfyUI_windows_portable\ComfyUI\models"
) else (
	echo ComfyUI folder not detected, give ComfyUI folder path:
    set /p "comfyPath=Path: "
    set "modelsPath=!comfyPath!\models"
    echo comfyPath = !comfyPath!
    echo modelsPath = !modelsPath!
    pause
)

:CHOOSE_OPTION
REM Ask user for installation type
echo Do you want to download WAN models?
echo A) bf16
echo B) fp16
echo C) fp8
echo D) All
echo E) No
set /p "CHOICE=Enter your choice (A,B,C,D or E) and press Enter: "
	if /i "%CHOICE%"=="A" (
	set "DOWNLOAD=yes"
) else if /i "%CHOICE%"=="B" (
	set "DOWNLOAD=yes"
) else if /i "%CHOICE%"=="C" (
	set "DOWNLOAD=yes"
) else if /i "%CHOICE%"=="D" (
	set "DOWNLOAD=yes"
) else if /i "%CHOICE%"=="E" (
	set "DOWNLOAD=no"
) else (
	echo Invalid choice. Please enter A,B,C,D or E.
	goto CHOOSE_OPTION
)

:CHOOSE_WAN_GGUF
REM Ask user if they want to download WAN GGUF Model
echo Do you want to download WAN text to video GGUF models?
echo A) Q8_0 (24GB Vram)
echo B) Q5_K_M (16GB Vram)
echo C) Q3_K_S (less than 12GB Vram)
echo D) All
echo E) No
set /p "WAN_GGUF_CHOICE=Enter your choice (A,B,C,D or E) and press Enter: "
	if /i "%WAN_GGUF_CHOICE%"=="A" (
	set "DOWNLOAD_GGUF=yes"
) else if /i "%WAN_GGUF_CHOICE%"=="B" (
	set "DOWNLOAD_GGUF=yes"
) else if /i "%WAN_GGUF_CHOICE%"=="C" (
	set "DOWNLOAD_GGUF=yes"
) else if /i "%WAN_GGUF_CHOICE%"=="D" (
	set "DOWNLOAD_GGUF=yes"
) else if /i "%WAN_GGUF_CHOICE%"=="E" (
	set "DOWNLOAD_GGUF=no"
) else (
	echo Invalid choice. Please enter A,B,C,D or E.
	goto CHOOSE_WAN_GGUF
)

:CHOOSE_WAN_GGUF_480
REM Ask user if they want to download WAN GGUF Model
echo Do you want to download WAN image to video 480p GGUF models?
echo A) Q8_0 (24GB Vram)
echo B) Q5_K_M (16GB Vram)
echo C) Q3_K_S (less than 12GB Vram)
echo D) All
echo E) No
set /p "WAN_GGUF_CHOICE_480=Enter your choice (A,B,C,D or E) and press Enter: "
	if /i "%WAN_GGUF_CHOICE_480%"=="A" (
	set "DOWNLOAD_GGUF_480=yes"
) else if /i "%WAN_GGUF_CHOICE_480%"=="B" (
	set "DOWNLOAD_GGUF_480=yes"
) else if /i "%WAN_GGUF_CHOICE_480%"=="C" (
	set "DOWNLOAD_GGUF_480=yes"
) else if /i "%WAN_GGUF_CHOICE_480%"=="D" (
	set "DOWNLOAD_GGUF_480=yes"
) else if /i "%WAN_GGUF_CHOICE_480%"=="E" (
	set "DOWNLOAD_GGUF=no"
) else (
	echo Invalid choice. Please enter A,B,C,D or E.
	goto CHOOSE_WAN_GGUF_480
)

:CHOOSE_WAN_GGUF_720
REM Ask user if they want to download WAN GGUF Model
echo Do you want to download WAN image to video 720p GGUF models?
echo A) Q8_0 (24GB Vram)
echo B) Q5_K_M (16GB Vram)
echo C) Q3_K_S (less than 12GB Vram)
echo D) All
echo E) No
set /p "WAN_GGUF_CHOICE_720=Enter your choice (A,B,C,D or E) and press Enter: "
	if /i "%WAN_GGUF_CHOICE_720%"=="A" (
	set "DOWNLOAD_GGUF_720=yes"
) else if /i "%WAN_GGUF_CHOICE_720%"=="B" (
	set "DOWNLOAD_GGUF_720=yes"
) else if /i "%WAN_GGUF_CHOICE_720%"=="C" (
	set "DOWNLOAD_GGUF_720=yes"
) else if /i "%WAN_GGUF_CHOICE_720%"=="D" (
	set "DOWNLOAD_GGUF_720=yes"
) else if /i "%WAN_GGUF_CHOICE_720%"=="E" (
	set "DOWNLOAD_GGUF_720=no"
) else (
	echo Invalid choice. Please enter A,B,C,D or E.
	goto CHOOSE_WAN_GGUF_720
)

:CHOOSE_WAN_FUN_CONTROL
REM Ask user if they want to download WAN GGUF Model
echo Do you want to download WAN FUN CONTROL base models?
echo A) bf16 (24GB Vram)
echo B) fp8 (16GB Vram)
echo C) All
echo D) No
set /p "WAN_CONTROL_CHOICE=Enter your choice (A,B,C or D) and press Enter: "
	if /i "%WAN_CONTROL_CHOICE%"=="A" (
	set "DOWNLOAD_CONTROL=yes"
) else if /i "%WAN_CONTROL_CHOICE%"=="B" (
	set "DOWNLOAD_CONTROL=yes"
) else if /i "%WAN_CONTROL_CHOICE%"=="C" (
	set "DOWNLOAD_CONTROL=yes"
) else if /i "%WAN_CONTROL_CHOICE%"=="D" (
	set "DOWNLOAD_CONTROL=yes"
) else (
	echo Invalid choice. Please enter A,B,C or D.
	goto CHOOSE_WAN_FUN_CONTROL
)

:CHOOSE_WAN_FUN_CONTROL_GGUF
REM Ask user if they want to download WAN GGUF Model
echo Do you want to download WAN FUN CONTROL GGUF models?
echo A) Q8_0 (24GB Vram)
echo B) Q5_K_M (16GB Vram)
echo C) Q3_K_S (less than 12GB Vram)
echo D) All
echo E) No
set /p "WAN_CONTROL_CHOICE_GGUF=Enter your choice (A,B,C,D or E) and press Enter: "
	if /i "%WAN_CONTROL_CHOICE_GGUF%"=="A" (
	set "DOWNLOAD_CONTROL_GGUF=yes"
) else if /i "%WAN_CONTROL_CHOICE_GGUF%"=="B" (
	set "DOWNLOAD_CONTROL_GGUF=yes"
) else if /i "%WAN_CONTROL_CHOICE_GGUF%"=="C" (
	set "DOWNLOAD_CONTROL_GGUF=yes"
) else if /i "%WAN_CONTROL_CHOICE_GGUF%"=="D" (
	set "DOWNLOAD_CONTROL_GGUF=yes"
) else if /i "%WAN_CONTROL_CHOICE_GGUF%"=="E" (
	set "DOWNLOAD_CONTROL_GGUF=no"
) else (
	echo Invalid choice. Please enter A,B,C,D or E.
	goto CHOOSE_WAN_FUN_CONTROL_GGUF
)

:CHOOSE_WAN_VACE
REM Ask user if they want to download WAN GGUF Model
echo Do you want to download WAN VACE base models?
echo A) fp16 (24GB Vram)
echo B) fp8 (16GB Vram)
echo C) All
echo D) No
set /p "WAN_VACE_CHOICE=Enter your choice (A,B,C or D) and press Enter: "
	if /i "%WAN_VACE_CHOICE%"=="A" (
	set "DOWNLOAD_VACE=yes"
) else if /i "%WAN_VACE_CHOICE%"=="B" (
	set "DOWNLOAD_VACE=yes"
) else if /i "%WAN_VACE_CHOICE%"=="C" (
	set "DOWNLOAD_VACE=yes"
) else if /i "%WAN_VACE_CHOICE%"=="D" (
	set "DOWNLOAD_VACE=yes"
) else (
	echo Invalid choice. Please enter A,B,C or D.
	goto CHOOSE_WAN_VACE
)

:CHOOSE_WAN_VACE_GGUF
REM Ask user if they want to download WAN GGUF Model
echo Do you want to download WAN VACE GGUF models?
echo A) Q8_0 (24GB Vram)
echo B) Q5_K_M (16GB Vram)
echo C) Q4_K_S (less than 12GB Vram)
echo D) All
echo E) No
set /p "WAN_VACE_CHOICE_GGUF=Enter your choice (A,B,C,D or E) and press Enter: "
	if /i "%WAN_VACE_CHOICE_GGUF%"=="A" (
	set "DOWNLOAD_VACE_GGUF=yes"
) else if /i "%WAN_VACE_CHOICE_GGUF%"=="B" (
	set "DOWNLOAD_VACE_GGUF=yes"
) else if /i "%WAN_VACE_CHOICE_GGUF%"=="C" (
	set "DOWNLOAD_VACE_GGUF=yes"
) else if /i "%WAN_VACE_CHOICE_GGUF%"=="D" (
	set "DOWNLOAD_VACE_GGUF=yes"
) else if /i "%WAN_VACE_CHOICE_GGUF%"=="E" (
	set "DOWNLOAD_VACE_GGUF=no"
) else (
	echo Invalid choice. Please enter A,B,C,D or E.
	goto CHOOSE_WAN_VACE_GGUF
)

if "%DOWNLOAD%"=="yes" (
    echo Downloading diffusion models file...
	powershell -NoProfile -Command "$p = Join-Path $env:modelsPath 'diffusion_models\WAN'; if (-Not (Test-Path $p)) { New-Item -ItemType Directory -Path $p | Out-Null }"
    if /i "%CHOICE%"=="A" (
	echo T2V bf16 Model :
	curl -L -o "%modelsPath%/diffusion_models/WAN/wan2.1_t2v_14B_bf16.safetensors" https://huggingface.co/UmeAiRT/ComfyUI-Auto_installer/resolve/main/models/diffusion_models/WAN/wan2.1_t2v_14B_bf16.safetensors?download=true
	curl -L -o "%modelsPath%/diffusion_models/WAN/wan2.1_t2v_1.3B_bf16.safetensors" https://huggingface.co/UmeAiRT/ComfyUI-Auto_installer/resolve/main/models/diffusion_models/WAN/wan2.1_t2v_1.3B_bf16.safetensors?download=true
	echo I2V bf16 Model :
	curl -L -o "%modelsPath%/diffusion_models/WAN/wan2.1_i2v_720p_14B_bf16.safetensors" https://huggingface.co/UmeAiRT/ComfyUI-Auto_installer/resolve/main/models/diffusion_models/WAN/wan2.1_i2v_720p_14B_bf16.safetensors?download=true
	curl -L -o "%modelsPath%/diffusion_models/WAN/wan2.1_i2v_480p_14B_bf16.safetensors" https://huggingface.co/UmeAiRT/ComfyUI-Auto_installer/resolve/main/models/diffusion_models/WAN/wan2.1_i2v_480p_14B_bf16.safetensors?download=true
	) else if /i "%CHOICE%"=="B" (
	echo T2V fp16 Model :
	curl -L -o "%modelsPath%/diffusion_models/WAN/wan2.1_t2v_14B_fp16.safetensors" https://huggingface.co/UmeAiRT/ComfyUI-Auto_installer/resolve/main/models/diffusion_models/WAN/wan2.1_t2v_14B_fp16.safetensors?download=true
	curl -L -o "%modelsPath%/diffusion_models/WAN/wan2.1_t2v_1.3B_fp16.safetensors" https://huggingface.co/UmeAiRT/ComfyUI-Auto_installer/resolve/main/models/diffusion_models/WAN/wan2.1_t2v_1.3B_fp16.safetensors?download=true
	echo I2V fp16 Model :
	curl -L -o "%modelsPath%/diffusion_models/WAN/wan2.1_i2v_720p_14B_fp16.safetensors" https://huggingface.co/UmeAiRT/ComfyUI-Auto_installer/resolve/main/models/diffusion_models/WAN/wan2.1_i2v_720p_14B_fp16.safetensors?download=true
	curl -L -o "%modelsPath%/diffusion_models/WAN/wan2.1_i2v_480p_14B_fp16.safetensors" https://huggingface.co/UmeAiRT/ComfyUI-Auto_installer/resolve/main/models/diffusion_models/WAN/wan2.1_i2v_480p_14B_fp16.safetensors?download=true
	) else if /i "%CHOICE%"=="C" (
	echo T2V fp8 Model :
	curl -L -o "%modelsPath%/diffusion_models/WAN/wan2.1_t2v_14B_fp8_e4m3fn.safetensors" https://huggingface.co/UmeAiRT/ComfyUI-Auto_installer/resolve/main/models/diffusion_models/WAN/wan2.1_t2v_14B_fp8_e4m3fn.safetensors?download=true
	curl -L -o "%modelsPath%/diffusion_models/WAN/wan2.1_t2v_1.3B_fp8_e4m3fn.safetensors" https://huggingface.co/UmeAiRT/ComfyUI-Auto_installer/resolve/main/models/diffusion_models/WAN/wan2.1_t2v_1.3B_fp8_e4m3fn.safetensors?download=true
	echo I2V fp8 Model :
	curl -L -o "%modelsPath%/diffusion_models/WAN/wan2.1_i2v_720p_14B_fp8_e4m3fn.safetensors" https://huggingface.co/UmeAiRT/ComfyUI-Auto_installer/resolve/main/models/diffusion_models/WAN/wan2.1_i2v_720p_14B_fp8_e4m3fn.safetensors?download=true
	curl -L -o "%modelsPath%/diffusion_models/WAN/wan2.1_i2v_480p_14B_fp8_e4m3fn.safetensors" https://huggingface.co/UmeAiRT/ComfyUI-Auto_installer/resolve/main/models/diffusion_models/WAN/wan2.1_i2v_480p_14B_fp8_e4m3fn.safetensors?download=true
	) else if /i "%CHOICE%"=="D" (
	
	echo T2V bf16 Model :
	curl -L -o "%modelsPath%/diffusion_models/WAN/wan2.1_t2v_14B_bf16.safetensors" https://huggingface.co/UmeAiRT/ComfyUI-Auto_installer/resolve/main/models/diffusion_models/WAN/wan2.1_t2v_14B_bf16.safetensors?download=true
	curl -L -o "%modelsPath%/diffusion_models/WAN/wan2.1_t2v_1.3B_bf16.safetensors" https://huggingface.co/UmeAiRT/ComfyUI-Auto_installer/resolve/main/models/diffusion_models/WAN/wan2.1_t2v_1.3B_bf16.safetensors?download=true
	echo I2V bf16 Model :
	curl -L -o "%modelsPath%/diffusion_models/WAN/wan2.1_i2v_720p_14B_bf16.safetensors" https://huggingface.co/UmeAiRT/ComfyUI-Auto_installer/resolve/main/models/diffusion_models/WAN/wan2.1_i2v_720p_14B_bf16.safetensors?download=true
	curl -L -o "%modelsPath%/diffusion_models/WAN/wan2.1_i2v_480p_14B_bf16.safetensors" https://huggingface.co/UmeAiRT/ComfyUI-Auto_installer/resolve/main/models/diffusion_models/WAN/wan2.1_i2v_480p_14B_bf16.safetensors?download=true
	echo T2V fp16 Model :
	curl -L -o "%modelsPath%/diffusion_models/WAN/wan2.1_t2v_14B_fp16.safetensors" https://huggingface.co/UmeAiRT/ComfyUI-Auto_installer/resolve/main/models/diffusion_models/WAN/wan2.1_t2v_14B_fp16.safetensors?download=true
	curl -L -o "%modelsPath%/diffusion_models/WAN/wan2.1_t2v_1.3B_fp16.safetensors" https://huggingface.co/UmeAiRT/ComfyUI-Auto_installer/resolve/main/models/diffusion_models/WAN/wan2.1_t2v_1.3B_fp16.safetensors?download=true
	echo I2V fp16 Model :
	curl -L -o "%modelsPath%/diffusion_models/WAN/wan2.1_i2v_720p_14B_fp16.safetensors" https://huggingface.co/UmeAiRT/ComfyUI-Auto_installer/resolve/main/models/diffusion_models/WAN/wan2.1_i2v_720p_14B_fp16.safetensors?download=true
	curl -L -o "%modelsPath%/diffusion_models/WAN/wan2.1_i2v_480p_14B_fp16.safetensors" https://huggingface.co/UmeAiRT/ComfyUI-Auto_installer/resolve/main/models/diffusion_models/WAN/wan2.1_i2v_480p_14B_fp16.safetensors?download=true
	echo T2V fp8 Model :
	curl -L -o "%modelsPath%/diffusion_models/WAN/wan2.1_t2v_14B_fp8_e4m3fn.safetensors" https://huggingface.co/UmeAiRT/ComfyUI-Auto_installer/resolve/main/models/diffusion_models/WAN/wan2.1_t2v_14B_fp8_e4m3fn.safetensors?download=true
	curl -L -o "%modelsPath%/diffusion_models/WAN/wan2.1_t2v_1.3B_fp8_e4m3fn.safetensors" https://huggingface.co/UmeAiRT/ComfyUI-Auto_installer/resolve/main/models/diffusion_models/WAN/wan2.1_t2v_1.3B_fp8_e4m3fn.safetensors?download=true
	echo I2V fp8 Model :
	curl -L -o "%modelsPath%/diffusion_models/WAN/wan2.1_i2v_720p_14B_fp8_e4m3fn.safetensors" https://huggingface.co/UmeAiRT/ComfyUI-Auto_installer/resolve/main/models/diffusion_models/WAN/wan2.1_i2v_720p_14B_fp8_e4m3fn.safetensors?download=true
	curl -L -o "%modelsPath%/diffusion_models/WAN/wan2.1_i2v_480p_14B_fp8_e4m3fn.safetensors" https://huggingface.co/UmeAiRT/ComfyUI-Auto_installer/resolve/main/models/diffusion_models/WAN/wan2.1_i2v_480p_14B_fp8_e4m3fn.safetensors?download=true
	)
)

REM Download VAE file
echo Downloading VAE file...
curl -L -o "%modelsPath%/vae/wan_2.1_vae.safetensors" https://huggingface.co/UmeAiRT/ComfyUI-Auto_installer/resolve/main/models/vae/wan_2.1_vae.safetensors?download=true

REM Download CLIP files
echo Downloading CLIP files...
curl -L -o "%modelsPath%/clip/umt5_xxl_fp8_e4m3fn_scaled.safetensors" https://huggingface.co/UmeAiRT/ComfyUI-Auto_installer/resolve/main/models/clip/umt5_xxl_fp8_e4m3fn_scaled.safetensors?download=true

if "%DOWNLOAD_GGUF%"=="yes" (
    echo Downloading GGUF T2V Quant Model...
	powershell -NoProfile -Command "$p = Join-Path $env:modelsPath 'unet\WAN'; if (-Not (Test-Path $p)) { New-Item -ItemType Directory -Path $p | Out-Null }"
    if /i "%WAN_GGUF_CHOICE%"=="A" (
	curl -L -o "%modelsPath%/unet/WAN/wan2.1-t2v-14b-Q8_0.gguf" https://huggingface.co/UmeAiRT/ComfyUI-Auto_installer/resolve/main/models/unet/WAN/wan2.1-t2v-14b-Q8_0.gguf?download=true
	) else if /i "%WAN_GGUF_CHOICE%"=="B" (
	curl -L -o "%modelsPath%/unet/WAN/wan2.1-t2v-14b-Q5_K_M.gguf" https://huggingface.co/UmeAiRT/ComfyUI-Auto_installer/resolve/main/models/unet/WAN/wan2.1-t2v-14b-Q5_K_M.gguf?download=true
	) else if /i "%WAN_GGUF_CHOICE%"=="C" (
	curl -L -o "%modelsPath%/unet/WAN/wan2.1-t2v-14b-Q3_K_S.gguf" https://huggingface.co/UmeAiRT/ComfyUI-Auto_installer/resolve/main/models/unet/WAN/wan2.1-t2v-14b-Q3_K_S.gguf?download=true
	) else if /i "%WAN_GGUF_CHOICE%"=="D" (
	curl -L -o "%modelsPath%/unet/WAN/wan2.1-t2v-14b-Q8_0.gguf" https://huggingface.co/UmeAiRT/ComfyUI-Auto_installer/resolve/main/models/unet/WAN/wan2.1-t2v-14b-Q8_0.gguf?download=true
	curl -L -o "%modelsPath%/unet/WAN/wan2.1-t2v-14b-Q5_K_M.gguf" https://huggingface.co/UmeAiRT/ComfyUI-Auto_installer/resolve/main/models/unet/WAN/wan2.1-t2v-14b-Q5_K_M.gguf?download=true
	curl -L -o "%modelsPath%/unet/WAN/wan2.1-t2v-14b-Q3_K_S.gguf" https://huggingface.co/UmeAiRT/ComfyUI-Auto_installer/resolve/main/models/unet/WAN/wan2.1-t2v-14b-Q3_K_S.gguf?download=true
	)
)
if "%DOWNLOAD_GGUF_480%"=="yes" (
    echo Downloading 480p GGUF I2V Quant Model...
	powershell -NoProfile -Command "$p = Join-Path $env:modelsPath 'unet\WAN'; if (-Not (Test-Path $p)) { New-Item -ItemType Directory -Path $p | Out-Null }"
    if /i "%WAN_GGUF_CHOICE_480%"=="A" (
	curl -L -o "%modelsPath%/unet/WAN/wan2.1-i2v-14b-480p-Q8_0.gguf" https://huggingface.co/UmeAiRT/ComfyUI-Auto_installer/resolve/main/models/unet/WAN/wan2.1-i2v-14b-480p-Q8_0.gguf?download=true
	) else if /i "%WAN_GGUF_CHOICE_480%"=="B" (
	curl -L -o "%modelsPath%/unet/WAN/wan2.1-i2v-14b-480p-Q5_K_M.gguf" https://huggingface.co/UmeAiRT/ComfyUI-Auto_installer/resolve/main/models/unet/WAN/wan2.1-i2v-14b-480p-Q5_K_M.gguf?download=true
	) else if /i "%WAN_GGUF_CHOICE_480%"=="C" (
	curl -L -o "%modelsPath%/unet/WAN/wan2.1-i2v-14b-480p-Q3_K_S.gguf" https://huggingface.co/UmeAiRT/ComfyUI-Auto_installer/resolve/main/models/unet/WAN/wan2.1-i2v-14b-480p-Q3_K_S.gguf?download=true
	) else if /i "%WAN_GGUF_CHOICE_480%"=="D" (
	curl -L -o "%modelsPath%/unet/WAN/wan2.1-i2v-14b-480p-Q8_0.gguf" https://huggingface.co/UmeAiRT/ComfyUI-Auto_installer/resolve/main/models/unet/WAN/wan2.1-i2v-14b-480p-Q8_0.gguf?download=true
	curl -L -o "%modelsPath%/unet/WAN/wan2.1-i2v-14b-480p-Q5_K_M.gguf" https://huggingface.co/UmeAiRT/ComfyUI-Auto_installer/resolve/main/models/unet/WAN/wan2.1-i2v-14b-480p-Q5_K_M.gguf?download=true
	curl -L -o "%modelsPath%/unet/WAN/wan2.1-i2v-14b-480p-Q3_K_S.gguf" https://huggingface.co/UmeAiRT/ComfyUI-Auto_installer/resolve/main/models/unet/WAN/wan2.1-i2v-14b-480p-Q3_K_S.gguf?download=true
	)
)

if "%DOWNLOAD_GGUF_720%"=="yes" (
    echo Downloading 720p GGUF I2V Quant Model...
	powershell -NoProfile -Command "$p = Join-Path $env:modelsPath 'unet\WAN'; if (-Not (Test-Path $p)) { New-Item -ItemType Directory -Path $p | Out-Null }"
    if /i "%WAN_GGUF_CHOICE%"=="A" (
	curl -L -o "%modelsPath%/unet/WAN/wan2.1-i2v-14b-720p-Q8_0.gguf" https://huggingface.co/UmeAiRT/ComfyUI-Auto_installer/resolve/main/models/unet/WAN/wan2.1-i2v-14b-720p-Q8_0.gguf?download=true
	) else if /i "%WAN_GGUF_CHOICE%"=="B" (
	curl -L -o "%modelsPath%/unet/WAN/wan2.1-i2v-14b-720p-Q5_K_M.gguf" https://huggingface.co/UmeAiRT/ComfyUI-Auto_installer/resolve/main/models/unet/WAN/wan2.1-i2v-14b-720p-Q5_K_M.gguf?download=true
	) else if /i "%WAN_GGUF_CHOICE%"=="C" (
	curl -L -o "%modelsPath%/unet/WAN/wan2.1-i2v-14b-720p-Q3_K_S.gguf" https://huggingface.co/UmeAiRT/ComfyUI-Auto_installer/resolve/main/models/unet/WAN/wan2.1-i2v-14b-720p-Q3_K_S.gguf?download=true
	) else if /i "%WAN_GGUF_CHOICE%"=="D" (
	curl -L -o "%modelsPath%/unet/WAN/wan2.1-i2v-14b-720p-Q8_0.gguf" https://huggingface.co/UmeAiRT/ComfyUI-Auto_installer/resolve/main/models/unet/WAN/wan2.1-i2v-14b-720p-Q8_0.gguf?download=true
	curl -L -o "%modelsPath%/unet/WAN/wan2.1-i2v-14b-720p-Q5_K_M.gguf" https://huggingface.co/UmeAiRT/ComfyUI-Auto_installer/resolve/main/models/unet/WAN/wan2.1-i2v-14b-720p-Q5_K_M.gguf?download=true
	curl -L -o "%modelsPath%/unet/WAN/wan2.1-i2v-14b-720p-Q3_K_S.gguf" https://huggingface.co/UmeAiRT/ComfyUI-Auto_installer/resolve/main/models/unet/WAN/wan2.1-i2v-14b-720p-Q3_K_S.gguf?download=true
	)
)

if "%DOWNLOAD_CONTROL%"=="yes" (
    echo Downloading FUN CONTROL Model...
	powershell -NoProfile -Command "$p = Join-Path $env:modelsPath 'unet\WAN'; if (-Not (Test-Path $p)) { New-Item -ItemType Directory -Path $p | Out-Null }"
    if /i "%WAN_VACE_CHOICE%"=="A" (
	curl -L -o "%modelsPath%/diffusion_models/WAN/wan2.1-fun-14B-control.safetensors" https://huggingface.co/UmeAiRT/ComfyUI-Auto_installer/resolve/main/models/diffusion_models/WAN/wan2.1-fun-14B-control.safetensors?download=true
	) else if /i "%WAN_VACE_CHOICE%"=="B" (
	curl -L -o "%modelsPath%/diffusion_models/WAN/Wan2.1-Fun-V1_1-InP-14B_fp8_e4m3fn.safetensors" https://huggingface.co/TFMC/Wan2.1-Fun-V1.1-14B-InP-FP8/resolve/main/Wan2.1-Fun-V1_1-InP-14B_fp8_e4m3fn.safetensors?download=true
	) else if /i "%WAN_VACE_CHOICE%"=="C" (
	curl -L -o "%modelsPath%/diffusion_models/WAN/wan2.1-fun-14B-control.safetensors" https://huggingface.co/UmeAiRT/ComfyUI-Auto_installer/resolve/main/models/diffusion_models/WAN/wan2.1-fun-14B-control.safetensors?download=true
	curl -L -o "%modelsPath%/diffusion_models/WAN/Wan2.1-Fun-V1_1-InP-14B_fp8_e4m3fn.safetensors" https://huggingface.co/TFMC/Wan2.1-Fun-V1.1-14B-InP-FP8/resolve/main/Wan2.1-Fun-V1_1-InP-14B_fp8_e4m3fn.safetensors?download=true
	)
)

if "%DOWNLOAD_CONTROL_GGUF%"=="yes" (
    echo Downloading FUN CONTROL GGUF Quant Model...
	powershell -NoProfile -Command "$p = Join-Path $env:modelsPath 'unet\WAN'; if (-Not (Test-Path $p)) { New-Item -ItemType Directory -Path $p | Out-Null }"
    if /i "%WAN_GGUF_CHOICE%"=="A" (
	curl -L -o "%modelsPath%/unet/WAN/wan2.1-fun-14b-control-Q8_0.gguf" https://huggingface.co/UmeAiRT/ComfyUI-Auto_installer/resolve/main/models/unet/WAN/wan2.1-fun-14b-control-Q8_0.gguf?download=true
	) else if /i "%WAN_GGUF_CHOICE%"=="B" (
	curl -L -o "%modelsPath%/unet/WAN/wan2.1-fun-14b-control-Q5_K_M.gguf" https://huggingface.co/UmeAiRT/ComfyUI-Auto_installer/resolve/main/models/unet/WAN/wan2.1-fun-14b-control-Q5_K_M.gguf?download=true
	) else if /i "%WAN_GGUF_CHOICE%"=="C" (
	curl -L -o "%modelsPath%/unet/WAN/wan2.1-fun-14b-control-Q3_K_S.gguf" https://huggingface.co/UmeAiRT/ComfyUI-Auto_installer/resolve/main/models/unet/WAN/wan2.1-fun-14b-control-Q3_K_S.gguf?download=true
	) else if /i "%WAN_GGUF_CHOICE%"=="D" (
	curl -L -o "%modelsPath%/unet/WAN/wan2.1-fun-14b-control-Q8_0.gguf" https://huggingface.co/UmeAiRT/ComfyUI-Auto_installer/resolve/main/models/unet/WAN/wan2.1-fun-14b-control-Q8_0.gguf?download=true
	curl -L -o "%modelsPath%/unet/WAN/wan2.1-fun-14b-control-Q5_K_M.gguf" https://huggingface.co/UmeAiRT/ComfyUI-Auto_installer/resolve/main/models/unet/WAN/wan2.1-fun-14b-control-Q5_K_M.gguf?download=true
	curl -L -o "%modelsPath%/unet/WAN/wan2.1-fun-14b-control-Q3_K_S.gguf" https://huggingface.co/UmeAiRT/ComfyUI-Auto_installer/resolve/main/models/unet/WAN/wan2.1-fun-14b-control-Q3_K_S.gguf?download=true
	)
)

if "%DOWNLOAD_VACE%"=="yes" (
    echo Downloading VACE Model...
	powershell -NoProfile -Command "$p = Join-Path $env:modelsPath 'unet\WAN'; if (-Not (Test-Path $p)) { New-Item -ItemType Directory -Path $p | Out-Null }"
    if /i "%WAN_VACE_CHOICE%"=="A" (
	curl -L -o "%modelsPath%/diffusion_models/WAN/wan2.1_vace_14B_fp16.safetensors" https://huggingface.co/UmeAiRT/ComfyUI-Auto_installer/resolve/main/models/diffusion_models/WAN/wan2.1_vace_14B_fp16.safetensors?download=true
	) else if /i "%WAN_VACE_CHOICE%"=="B" (
	curl -L -o "%modelsPath%/diffusion_models/WAN/wan2.1_vace_14B_fp8_e4m3fn.safetensors" https://huggingface.co/Kamikaze-88/Wan2.1-VACE-14B-fp8/resolve/main/wan2.1_vace_14B_fp8_e4m3fn.safetensors?download=true
	) else if /i "%WAN_VACE_CHOICE%"=="C" (
	curl -L -o "%modelsPath%/diffusion_models/WAN/wan2.1_vace_14B_fp16.safetensors" https://huggingface.co/UmeAiRT/ComfyUI-Auto_installer/resolve/main/models/diffusion_models/WAN/wan2.1_vace_14B_fp16.safetensors?download=true
	curl -L -o "%modelsPath%/diffusion_models/WAN/wan2.1_vace_14B_fp8_e4m3fn.safetensors" https://huggingface.co/Kamikaze-88/Wan2.1-VACE-14B-fp8/resolve/main/wan2.1_vace_14B_fp8_e4m3fn.safetensors?download=true
	)
)

if "%DOWNLOAD_VACE_GGUF%"=="yes" (
    echo Downloading VACE GGUF Quant Model...
	powershell -NoProfile -Command "$p = Join-Path $env:modelsPath 'unet\WAN'; if (-Not (Test-Path $p)) { New-Item -ItemType Directory -Path $p | Out-Null }"
    if /i "%WAN_VACE_CHOICE_GGUF%"=="A" (
	curl -L -o "%modelsPath%/unet/WAN/Wan2.1-VACE-14B-Q8_0.gguf" https://huggingface.co/UmeAiRT/ComfyUI-Auto_installer/resolve/main/models/unet/WAN/Wan2.1-VACE-14B-Q8_0.gguf?download=true
	) else if /i "%WAN_VACE_CHOICE_GGUF%"=="B" (
	curl -L -o "%modelsPath%/unet/WAN/Wan2.1-VACE-14B-Q5_K_S.gguf" https://huggingface.co/UmeAiRT/ComfyUI-Auto_installer/resolve/main/models/unet/WAN/Wan2.1-VACE-14B-Q5_K_S.gguf?download=true
	) else if /i "%WAN_VACE_CHOICE_GGUF%"=="C" (
	curl -L -o "%modelsPath%/unet/WAN/wan2.1-fun-14b-control-Q3_K_S.gguf" https://huggingface.co/UmeAiRT/ComfyUI-Auto_installer/resolve/main/models/unet/WAN/wan2.1-fun-14b-control-Q3_K_S.gguf?download=true
	) else if /i "%WAN_VACE_CHOICE_GGUF%"=="D" (
	curl -L -o "%modelsPath%/unet/WAN/Wan2.1-VACE-14B-Q8_0.gguf" https://huggingface.co/UmeAiRT/ComfyUI-Auto_installer/resolve/main/models/unet/WAN/Wan2.1-VACE-14B-Q8_0.gguf?download=true
	curl -L -o "%modelsPath%/unet/WAN/Wan2.1-VACE-14B-Q5_K_S.gguf" https://huggingface.co/UmeAiRT/ComfyUI-Auto_installer/resolve/main/models/unet/WAN/Wan2.1-VACE-14B-Q5_K_S.gguf?download=true
	curl -L -o "%modelsPath%/unet/WAN/Wan2.1-VACE-14B-Q4_K_S.gguf" https://huggingface.co/UmeAiRT/ComfyUI-Auto_installer/resolve/main/models/unet/WAN/Wan2.1-VACE-14B-Q4_K_S.gguf?download=true
	)
)

REM Download clip vision
echo Downloading clip vision file...
curl -L -o "%modelsPath%/clip_vision/clip_vision_h.safetensors" https://huggingface.co/UmeAiRT/ComfyUI-Auto_installer/resolve/main/models/clip_vision/clip_vision_h.safetensors?download=true

REM Download upscale model
echo Downloading upscale models...
curl -L -o "%modelsPath%/upscale_models/RealESRGAN_x4plus.pth" https://huggingface.co/UmeAiRT/ComfyUI-Auto_installer/resolve/main/models/upscale_models/RealESRGAN_x4plus.pth?download=true
curl -L -o "%modelsPath%/upscale_models/RealESRGAN_x4plus_anime_6B.pth" https://huggingface.co/UmeAiRT/ComfyUI-Auto_installer/resolve/main/models/upscale_models/RealESRGAN_x4plus_anime_6B.pth?download=true

echo Models downloaded.
pause