Upload UmeAiRT-HIDREAM-Model_downloader.bat

scripts/Model_downloader/UmeAiRT-HIDREAM-Model_downloader.bat

@@ -0,0 +1,93 @@
+@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 HiDream fp8 models? (24GB Vram recommanded)
+echo A) Yes
+echo B) No
+set /p "CHOICE=Enter your choice (A or B) and press Enter: "
+if /i "%CHOICE%"=="A" (
+    echo Downloading unet models file...
+	powershell -NoProfile -Command "$p = Join-Path $env:modelsPath 'diffusion_models\HIDREAM'; if (-Not (Test-Path $p)) { New-Item -ItemType Directory -Path $p | Out-Null }"
+	curl -L -o "%modelsPath%/unet/HIDREAM/hidream_i1_dev_fp8.safetensors" https://huggingface.co/UmeAiRT/ComfyUI-Auto_installer/resolve/main/models/diffusion_models/HiDream/hidream_i1_dev_fp8.safetensors?download=true
+) else if /i "%CHOICE%"=="B" (
+    
+) else (
+echo Invalid choice. Please enter A,B,C or D.
+    goto CHOOSE_OPTION
+)
+
+:CHOOSE_HID_GGUF
+REM Ask user if they HIDt to download HID GGUF Model
+echo Do you HIDt to download HiDream GGUF models?
+echo A) Q8_0 (16GB Vram)
+echo B) Q5_K_S (12GB Vram)
+echo C) Q4_K_S (less than 12GB Vram)
+echo D) All
+echo E) No
+set /p "HID_GGUF_CHOICE=Enter your choice (A,B,C,D or E) and press Enter: "
+	if /i "%HID_GGUF_CHOICE%"=="A" (
+	set "DOWNLOAD_GGUF=yes"
+) else if /i "%HID_GGUF_CHOICE%"=="B" (
+	set "DOWNLOAD_GGUF=yes"
+) else if /i "%HID_GGUF_CHOICE%"=="C" (
+	set "DOWNLOAD_GGUF=yes"
+) else if /i "%HID_GGUF_CHOICE%"=="D" (
+	set "DOWNLOAD_GGUF=yes"
+) else if /i "%HID_GGUF_CHOICE%"=="E" (
+	set "DOWNLOAD_GGUF=no"
+) else (
+	echo Invalid choice. Please enter A,B,C,D or E.
+	goto CHOOSE_HID_GGUF
+)
+
+REM Download VAE file
+echo Downloading VAE file...
+curl -L -o "%modelsPath%/vae/ae.safetensors" "https://huggingface.co/UmeAiRT/ComfyUI-Auto_installer/resolve/main/models/vae/ae.safetensors?download=true"
+
+REM Download CLIP files
+echo Downloading CLIP files...
+curl -L -o "%modelsPath%/clip/clip_g_hidream.safetensors" https://huggingface.co/UmeAiRT/ComfyUI-Auto_installer/resolve/main/models/clip/clip_g_hidream.safetensors?download=true
+curl -L -o "%modelsPath%/clip/clip_l_hidream.safetensors" https://huggingface.co/UmeAiRT/ComfyUI-Auto_installer/resolve/main/models/clip/clip_l_hidream.safetensors?download=true
+curl -L -o "%modelsPath%/clip/t5xxl_fp8_e4m3fn_scaled.safetensors" https://huggingface.co/UmeAiRT/ComfyUI-Auto_installer/resolve/main/models/clip/t5xxl_fp8_e4m3fn_scaled.safetensors?download=true
+curl -L -o "%modelsPath%/clip/llama_3.1_8b_instruct_fp8_scaled.safetensors" https://huggingface.co/UmeAiRT/ComfyUI-Auto_installer/resolve/main/models/clip/llama_3.1_8b_instruct_fp8_scaled.safetensors?download=true
+
+if "%DOWNLOAD_GGUF%"=="yes" (
+    echo Downloading HID GGUF Quant Model...
+	powershell -NoProfile -Command "$p = Join-Path $env:modelsPath 'unet\HIDREAM'; if (-Not (Test-Path $p)) { New-Item -ItemType Directory -Path $p | Out-Null }"
+    if /i "%HID_GGUF_CHOICE%"=="A" (
+	curl -L -o "%modelsPath%/unet/HIDREAM/hidream-i1-dev-Q8_0.gguf" https://huggingface.co/UmeAiRT/ComfyUI-Auto_installer/resolve/main/models/unet/HiDream/hidream-i1-dev-Q8_0.gguf?download=true
+	) else if /i "%HID_GGUF_CHOICE%"=="B" (
+	curl -L -o "%modelsPath%/unet/hidream-i1-dev-Q5_K_S.gguf" https://huggingface.co/UmeAiRT/ComfyUI-Auto_installer/resolve/main/models/unet/HiDream/hidream-i1-dev-Q5_K_S.gguf?download=true
+	) else if /i "%HID_GGUF_CHOICE%"=="C" (
+	curl -L -o "%modelsPath%/unet/hidream-i1-dev-Q4_K_S.gguf" https://huggingface.co/UmeAiRT/ComfyUI-Auto_installer/resolve/main/models/unet/HiDream/hidream-i1-dev-Q4_K_S.gguf?download=true
+	) else if /i "%HID_GGUF_CHOICE%"=="D" (
+	curl -L -o "%modelsPath%/unet/HIDREAM/hidream-i1-dev-Q8_0.gguf" https://huggingface.co/UmeAiRT/ComfyUI-Auto_installer/resolve/main/models/unet/HiDream/hidream-i1-dev-Q8_0.gguf?download=true
+	curl -L -o "%modelsPath%/unet/hidream-i1-dev-Q5_K_S.gguf" https://huggingface.co/UmeAiRT/ComfyUI-Auto_installer/resolve/main/models/unet/HiDream/hidream-i1-dev-Q5_K_S.gguf?download=true
+	curl -L -o "%modelsPath%/unet/hidream-i1-dev-Q4_K_S.gguf" https://huggingface.co/UmeAiRT/ComfyUI-Auto_installer/resolve/main/models/unet/HiDream/hidream-i1-dev-Q4_K_S.gguf?download=true
+	)
+)
+
+echo Models downloaded.
+pause