Upload UmeAiRT-FLUX-Model_downloader.bat

UmeAiRT-FLUX-Model_downloader.bat

@@ -87,6 +87,22 @@ echo Invalid choice. Please enter A or B.
     goto CHOOSE_FLUX_CONTROLNET
 )
 
+:CHOOSE_FLUX_PULID
+REM Ask user if they want to download FLUX SCHNELL Model
+echo Do you want to download FLUX PuLID and REDUX Models?
+echo A) Yes
+echo B) No
+set /p "FLUX_PULID_CHOICE=Enter your choice (A or B) and press Enter: "
+
+if /i "%FLUX_PULID_CHOICE%"=="A" (
+    set "DOWNLOAD_FLUX_PULID=yes"
+) else if /i "%FLUX_PULID_CHOICE%"=="B" (
+    set "DOWNLOAD_FLUX_PULID=no"
+) else (
+echo Invalid choice. Please enter A or B.
+    goto CHOOSE_FLUX_PULID
+)
+
 :CHOOSE_FLUX_LORA
 REM Ask user if they want to download FLUX LORA
 echo Do you want to download UmeAiRT LoRAs?
@@ -212,9 +228,14 @@ if "%DOWNLOAD_FLUX_LORA%"=="yes" (
     curl -L -o "%modelsPath%/loras/ume_knight.safetensors" https://huggingface.co/UmeAiRT/FLUX.1-dev-LoRA-Ume_Knight/resolve/main/ume_gachaak.safetensors?download=true
 )
 
-echo Downloading FLUX PuLID...
-mkdir "%modelsPath%/pulid"
-curl -L -o "%modelsPath%/pulid/pulid_flux_v0.9.0.safetensors" https://huggingface.co/camenduru/PuLID/resolve/main/pulid_flux_v0.9.0.safetensors?download=true
+REM Download FLUX SCHNELL Model if user chose to
+if "%DOWNLOAD_FLUX_PuLID%"=="yes" (
+    echo Downloading FLUX PuLID...
+	mkdir "%modelsPath%/pulid"
+    curl -L -o "%modelsPath%/pulid/pulid_flux_v0.9.0.safetensors" https://huggingface.co/camenduru/PuLID/resolve/main/pulid_flux_v0.9.0.safetensors?download=true
+    curl -L -o "%modelsPath%/clip_vision/sigclip_vision_patch14_384.safetensors" https://huggingface.co/Comfy-Org/sigclip_vision_384/resolve/main/sigclip_vision_patch14_384.safetensors?download=true
+    curl -L -o "%modelsPath%/style_models/flux1-redux-dev.safetensors" https://huggingface.co/black-forest-labs/FLUX.1-Redux-dev/resolve/main/flux1-redux-dev.safetensors?download=true
+)
 
 echo Models downloaded.
 pause