Upload UmeAiRT-WAN2.1-Model_downloader.bat

UmeAiRT-WAN2.1-Model_downloader.bat

@@ -23,43 +23,44 @@ if exist "%basePath%\ComfyUI" (
 
 :CHOOSE_OPTION
 REM Ask user for installation type
-echo Choose your preferred installation:
-echo A) Fast-Lowvram install (GGUF)
-echo B) Unoptimized normal model (bf16)
-set /p "CHOICE=Enter your choice (A or B) and press Enter: "
+echo Do you want to download WAN bf16 models?
+set /p "CHOICE=Enter your choice (Y or N) and press Enter: "
 
-if /i "%CHOICE%"=="A" (
-    set "INSTALL_TYPE=fast-lowvram"
-) else if /i "%CHOICE%"=="B" (
+if /i "%CHOICE%"=="Y" (
+    echo Downloading diffusion models file...
+	echo T2V Quant Model :
+	curl -L -o "%modelsPath%/diffusion_models/wan2.1_i2v_480p_14B_bf16.safetensors" https://huggingface.co/Comfy-Org/Wan_2.1_ComfyUI_repackaged/resolve/main/split_files/diffusion_models/wan2.1_i2v_480p_14B_bf16.safetensors?download=true
+	echo I2V Quant Model :
+	curl -L -o "%modelsPath%/diffusion_models/wan2.1_t2v_1.3B_bf16.safetensors" https://huggingface.co/Comfy-Org/Wan_2.1_ComfyUI_repackaged/resolve/main/split_files/diffusion_models/wan2.1_t2v_1.3B_bf16.safetensors?download=true
+) else if /i "%CHOICE%"=="N" (
     set "INSTALL_TYPE=unoptimized"
-	set "DOWNLOAD_GGUF=no"
 ) else (
 echo Invalid choice. Please enter A or B.
     goto CHOOSE_OPTION
 )
 
-:CHOOSE_FLUX_GGUF
-REM Ask user if they want to download FLUX GGUF Model
-echo Do you want to download FLUX GGUF Models?
+:CHOOSE_WAN_GGUF
+REM Ask user if they want to download WAN GGUF Model
+echo Do you want to download WAN 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 D) All
 echo E) No
-set /p "FLUX_GGUF_CHOICE=Enter your choice (A,B,C,D or E) and press Enter: "
-	if /i "%FLUX_GGUF_CHOICE%"=="A" (
+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 "%FLUX_GGUF_CHOICE%"=="B" (
+) else if /i "%WAN_GGUF_CHOICE%"=="B" (
 	set "DOWNLOAD_GGUF=yes"
-) else if /i "%FLUX_GGUF_CHOICE%"=="C" (
+) else if /i "%WAN_GGUF_CHOICE%"=="C" (
 	set "DOWNLOAD_GGUF=yes"
-) else if /i "%FLUX_GGUF_CHOICE%"=="D" (
+) else if /i "%WAN_GGUF_CHOICE%"=="D" (
 	set "DOWNLOAD_GGUF=yes"
-) else if /i "%FLUX_GGUF_CHOICE%"=="E" (
+) else if /i "%WAN_GGUF_CHOICE%"=="E" (
 	set "DOWNLOAD_GGUF=no"
 ) else (
 	echo Invalid choice. Please enter A or B.
-	goto CHOOSE_FLUX_GGUF
+	goto CHOOSE_WAN_GGUF
 )
 
 REM Download VAE file
@@ -70,23 +71,15 @@ REM Download CLIP files
 echo Downloading CLIP files...
 curl -L -o "%modelsPath%/clip/umt5_xxl_fp8_e4m3fn_scaled.safetensors" https://huggingface.co/Comfy-Org/Wan_2.1_ComfyUI_repackaged/resolve/main/split_files/text_encoders/umt5_xxl_fp8_e4m3fn_scaled.safetensors?download=true
 
-REM Download Diffusion models files
-if "%INSTALL_TYPE%"=="unoptimized" (
-	echo Downloading diffusion models file...
-	echo T2V Quant Model :
-	curl -L -o "%modelsPath%/diffusion_models/wan2.1_i2v_480p_14B_bf16.safetensors" https://huggingface.co/Comfy-Org/Wan_2.1_ComfyUI_repackaged/resolve/main/split_files/diffusion_models/wan2.1_i2v_480p_14B_bf16.safetensors?download=true
-	echo I2V Quant Model :
-	curl -L -o "%modelsPath%/diffusion_models/wan2.1_t2v_1.3B_bf16.safetensors" https://huggingface.co/Comfy-Org/Wan_2.1_ComfyUI_repackaged/resolve/main/split_files/diffusion_models/wan2.1_t2v_1.3B_bf16.safetensors?download=true
-)
 if "%DOWNLOAD_GGUF%"=="yes" (
     echo Downloading GGUF T2V Quant Model...
-    if /i "%FLUX_GGUF_CHOICE%"=="A" (
+    if /i "%WAN_GGUF_CHOICE%"=="A" (
 	curl -L -o "%modelsPath%/diffusion_models/wan2.1-t2v-14b-Q8_0.gguf" https://huggingface.co/city96/Wan2.1-T2V-14B-gguf/resolve/main/wan2.1-t2v-14b-Q8_0.gguf?download=true
-	) else if /i "%FLUX_GGUF_CHOICE%"=="B" (
+	) else if /i "%WAN_GGUF_CHOICE%"=="B" (
 	curl -L -o "%modelsPath%/diffusion_models/wan2.1-t2v-14b-Q5_K_M.gguf" https://huggingface.co/city96/Wan2.1-T2V-14B-gguf/resolve/main/wan2.1-t2v-14b-Q5_K_M.gguf?download=true
-	) else if /i "%FLUX_GGUF_CHOICE%"=="C" (
+	) else if /i "%WAN_GGUF_CHOICE%"=="C" (
 	curl -L -o "%modelsPath%/diffusion_models/wan2.1-t2v-14b-Q3_K_S.gguf" https://huggingface.co/city96/Wan2.1-T2V-14B-gguf/resolve/main/wan2.1-t2v-14b-Q3_K_S.gguf?download=true
-	) else if /i "%FLUX_GGUF_CHOICE%"=="D" (
+	) else if /i "%WAN_GGUF_CHOICE%"=="D" (
 	curl -L -o "%modelsPath%/diffusion_models/wan2.1-t2v-14b-Q8_0.gguf" https://huggingface.co/city96/Wan2.1-T2V-14B-gguf/resolve/main/wan2.1-t2v-14b-Q8_0.gguf?download=true
 	curl -L -o "%modelsPath%/diffusion_models/wan2.1-t2v-14b-Q5_K_M.gguf" https://huggingface.co/city96/Wan2.1-T2V-14B-gguf/resolve/main/wan2.1-t2v-14b-Q5_K_M.gguf?download=true
 	curl -L -o "%modelsPath%/diffusion_models/wan2.1-t2v-14b-Q3_K_S.gguf" https://huggingface.co/city96/Wan2.1-T2V-14B-gguf/resolve/main/wan2.1-t2v-14b-Q3_K_S.gguf?download=true
@@ -94,13 +87,13 @@ if "%DOWNLOAD_GGUF%"=="yes" (
 )
 if "%DOWNLOAD_GGUF%"=="yes" (
     echo Downloading GGUF I2V Quant Model...
-    if /i "%FLUX_GGUF_CHOICE%"=="A" (
+    if /i "%WAN_GGUF_CHOICE%"=="A" (
 	curl -L -o "%modelsPath%/diffusion_models/wan2.1-i2v-14b-480p-Q8_0.gguf" https://huggingface.co/city96/Wan2.1-I2V-14B-480P-gguf/resolve/main/wan2.1-i2v-14b-480p-Q8_0.gguf?download=true
-	) else if /i "%FLUX_GGUF_CHOICE%"=="B" (
+	) else if /i "%WAN_GGUF_CHOICE%"=="B" (
 	curl -L -o "%modelsPath%/diffusion_models/wan2.1-i2v-14b-480p-Q5_K_M.gguf" https://huggingface.co/city96/Wan2.1-I2V-14B-480P-gguf/resolve/main/wan2.1-i2v-14b-480p-Q5_K_M.gguf?download=true
-	) else if /i "%FLUX_GGUF_CHOICE%"=="C" (
+	) else if /i "%WAN_GGUF_CHOICE%"=="C" (
 	curl -L -o "%modelsPath%/diffusion_models/wan2.1-i2v-14b-480p-Q3_K_S.gguf" https://huggingface.co/city96/Wan2.1-I2V-14B-480P-gguf/resolve/main/wan2.1-i2v-14b-480p-Q3_K_S.gguf?download=true
-	) else if /i "%FLUX_GGUF_CHOICE%"=="D" (
+	) else if /i "%WAN_GGUF_CHOICE%"=="D" (
 	curl -L -o "%modelsPath%/diffusion_models/wan2.1-i2v-14b-480p-Q8_0.gguf" https://huggingface.co/city96/Wan2.1-I2V-14B-480P-gguf/resolve/main/wan2.1-i2v-14b-480p-Q8_0.gguf?download=true
 	curl -L -o "%modelsPath%/diffusion_models/wan2.1-i2v-14b-480p-Q5_K_M.gguf" https://huggingface.co/city96/Wan2.1-I2V-14B-480P-gguf/resolve/main/wan2.1-i2v-14b-480p-Q5_K_M.gguf?download=true
 	curl -L -o "%modelsPath%/diffusion_models/wan2.1-i2v-14b-480p-Q3_K_S.gguf" https://huggingface.co/city96/Wan2.1-I2V-14B-480P-gguf/resolve/main/wan2.1-i2v-14b-480p-Q3_K_S.gguf?download=true