Upload UmeAiRT-WAN2.1-Model_downloader.bat

scripts/Model_downloader/UmeAiRT-WAN2.1-Model_downloader.bat

@@ -119,13 +119,12 @@ set /p "WAN_GGUF_CHOICE_720=Enter your choice (A,B,C,D or E) and press Enter: "
 
 :CHOOSE_WAN_FUN_CONTROL
 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=Enter your choice (A,B,C,D or E) and press Enter: "
+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" (
@@ -134,13 +133,80 @@ set /p "WAN_CONTROL_CHOICE=Enter your choice (A,B,C,D or E) and press Enter: "
 	set "DOWNLOAD_CONTROL=yes"
 ) else if /i "%WAN_CONTROL_CHOICE%"=="D" (
 	set "DOWNLOAD_CONTROL=yes"
-) else if /i "%WAN_CONTROL_CHOICE%"=="E" (
-	set "DOWNLOAD_CONTROL=no"
 ) else (
-	echo Invalid choice. Please enter A,B,C,D or E.
+	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 }"
@@ -244,6 +310,19 @@ if "%DOWNLOAD_GGUF_720%"=="yes" (
 )
 
 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" (
@@ -259,6 +338,35 @@ if "%DOWNLOAD_CONTROL%"=="yes" (
 	)
 )
 
+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