Upload UmeAiRT-WAN2.1-Model_downloader.bat

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

@@ -48,9 +48,9 @@ set /p "CHOICE=Enter your choice (A,B,C,D or E) and press Enter: "
 :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 + T5_Q8 (24GB Vram)
-echo B) Q5_K_S + T5_Q5_K_M (16GB Vram)
-echo C) Q4_K_S + T5_Q3_K_L (less than 12GB Vram)
+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: "
@@ -72,9 +72,9 @@ set /p "WAN_GGUF_CHOICE=Enter your choice (A,B,C,D or E) and press Enter: "
 :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 + T5_Q8 (24GB Vram)
-echo B) Q5_K_S + T5_Q5_K_M (16GB Vram)
-echo C) Q4_K_S + T5_Q3_K_L (less than 12GB Vram)
+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: "
@@ -96,9 +96,9 @@ set /p "WAN_GGUF_CHOICE_480=Enter your choice (A,B,C,D or E) and press Enter: "
 :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 + T5_Q8 (24GB Vram)
-echo B) Q5_K_S + T5_Q5_K_M (16GB Vram)
-echo C) Q4_K_S + T5_Q3_K_L (less than 12GB Vram)
+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: "
@@ -117,6 +117,30 @@ set /p "WAN_GGUF_CHOICE_720=Enter your choice (A,B,C,D or E) and press Enter: "
 	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 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_720=Enter your choice (A,B,C,D or E) 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 if /i "%WAN_CONTROL_CHOICE%"=="E" (
+	set "DOWNLOAD_CONTROL=no"
+) else (
+	echo Invalid choice. Please enter A,B,C,D or E.
+	goto CHOOSE_WAN_FUN_CONTROL
+)
+
 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 }"
@@ -219,6 +243,22 @@ if "%DOWNLOAD_GGUF_720%"=="yes" (
 	)
 )
 
+if "%DOWNLOAD_CONTROL%"=="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
+	)
+)
+
 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