ComfyUI-AllinOne-Auto_install.bat

@echo off
setlocal enabledelayedexpansion

:CHOOSE_OPTION
REM Ask user for installation type
echo Choose your preferred installation:
echo A) Fast-Lowvram install
echo B) Unoptimized normal model
set /p "CHOICE=Enter your choice (A or B) and press Enter: "

if /i "%CHOICE%"=="A" (
    set "INSTALL_TYPE=fast-lowvram"
) else if /i "%CHOICE%"=="B" (
    set "INSTALL_TYPE=unoptimized"
) else (
echo Invalid choice. Please enter A or B.
    goto CHOOSE_OPTION
)

:CHOOSE_FLUX_SCHNELL
REM Ask user if they want to download FLUX SCHNELL Model
echo Do you want to download the FLUX SCHNELL Model?
echo A) Yes
echo B) No
set /p "FLUX_SCHNELL_CHOICE=Enter your choice (A or B) and press Enter: "

if /i "%FLUX_SCHNELL_CHOICE%"=="A" (
    set "DOWNLOAD_FLUX_SCHNELL=yes"
) else if /i "%FLUX_SCHNELL_CHOICE%"=="B" (
    set "DOWNLOAD_FLUX_SCHNELL=no"
) else (
echo Invalid choice. Please enter A or B.
    goto CHOOSE_FLUX_SCHNELL
)

:CHOOSE_FLUX_GGUF
REM Ask user if they want to download FLUX GGUF Model
echo Do you want to download FLUX 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 "DOWNLOAD_GGUF=yes"
) else if /i "%FLUX_GGUF_CHOICE%"=="B" (
    set "DOWNLOAD_GGUF=yes"
) else if /i "%FLUX_GGUF_CHOICE%"=="C" (
    set "DOWNLOAD_GGUF=yes"
) else if /i "%FLUX_GGUF_CHOICE%"=="D" (
    set "DOWNLOAD_GGUF=yes"
) else if /i "%FLUX_GGUF_CHOICE%"=="E" (
    set "DOWNLOAD_GGUF=no"
) else (
echo Invalid choice. Please enter A or B.
    goto CHOOSE_FLUX_GGUF
)

:CHOOSE_FLUX_CONTROLNET
REM Ask user if they want to download FLUX SCHNELL Model
echo Do you want to download ControlNet V3 Models?
echo A) Yes
echo B) No
set /p "FLUX_CONTROLNET_CHOICE=Enter your choice (A or B) and press Enter: "

if /i "%FLUX_CONTROLNET_CHOICE%"=="A" (
    set "DOWNLOAD_FLUX_CONTROLNET=yes"
) else if /i "%FLUX_CONTROLNET_CHOICE%"=="B" (
    set "DOWNLOAD_FLUX_CONTROLNET=no"
) else (
echo Invalid choice. Please enter A or B.
    goto CHOOSE_FLUX_CONTROLNET
)

:CHOOSE_FLUX_LORA
REM Ask user if they want to download FLUX LORA
echo Do you want to download UmeAiRT LoRAs?
echo A) Yes
echo B) No
set /p "FLUX_LORA_CHOICE=Enter your choice (A or B) and press Enter: "

if /i "%FLUX_LORA_CHOICE%"=="A" (
    set "DOWNLOAD_FLUX_LORA=yes"
) else if /i "%FLUX_LORA_CHOICE%"=="B" (
    set "DOWNLOAD_FLUX_LORA=no"
) else (
echo Invalid choice. Please enter A or B.
    goto CHOOSE_FLUX_LORA
)

REM Check if 7-Zip is installed and get its path
for %%I in (7z.exe) do set "SEVEN_ZIP_PATH=%%~$PATH:I"
if not defined SEVEN_ZIP_PATH (
    if exist "%ProgramFiles%\7-Zip\7z.exe" (
        set "SEVEN_ZIP_PATH=%ProgramFiles%\7-Zip\7z.exe"
    ) else if exist "%ProgramFiles(x86)%\7-Zip\7z.exe" (
        set "SEVEN_ZIP_PATH=%ProgramFiles(x86)%\7-Zip\7z.exe"
    ) else (
        echo 7-Zip is not installed. Downloading and installing...
        curl -L -o 7z-installer.exe https://www.7-zip.org/a/7z2201-x64.exe
        7z-installer.exe /S
        set "SEVEN_ZIP_PATH=%ProgramFiles%\7-Zip\7z.exe"
        if not exist "%SEVEN_ZIP_PATH%" (
            echo Installation of 7-Zip failed. Please install it manually and try again.
            exit /b 1
        )
        del 7z-installer.exe
    )
)

REM Check and install Git
git --version > NUL 2>&1
if %errorlevel% NEQ 0 (
    echo Installing Git...
    powershell -Command "& {Invoke-WebRequest -Uri 'https://github.com/git-for-windows/git/releases/download/v2.41.0.windows.3/Git-2.41.0.3-64-bit.exe' -OutFile 'Git-2.41.0.3-64-bit.exe'; if ($LASTEXITCODE -ne 0) { exit 1 }}"
    if %errorlevel% NEQ 0 (
        echo Failed to download Git installer.
        exit /b
    )
    start /wait Git-2.41.0.3-64-bit.exe /VERYSILENT
    del Git-2.41.0.3-64-bit.exe
) else (
    echo Git already installed.
)

REM Download ComfyUI
echo Downloading ComfyUI...
curl -L -o ComfyUI_windows_portable_nvidia_cu121_or_cpu.7z https://github.com/comfyanonymous/ComfyUI/releases/download/latest/ComfyUI_windows_portable_nvidia_cu121_or_cpu.7z

REM Extract ComfyUI
echo Extracting ComfyUI...
"%SEVEN_ZIP_PATH%" x ComfyUI_windows_portable_nvidia_cu121_or_cpu.7z -o"%CD%" -y

REM Check if extraction was successful
if not exist "ComfyUI_windows_portable" (
    echo Extraction failed. Please check the downloaded file and try again.
    exit /b 1
)

REM Delete archive
del /f ComfyUI_windows_portable_nvidia_cu121_or_cpu.7z -force


REM Navigate to custom_nodes folder
REM Update ComfyUI
cd ComfyUI_windows_portable\update
..\python_embeded\python.exe -m pip install --upgrade pip
..\python_embeded\python.exe .\update.py ..\ComfyUI\
if exist update_new.py (
  move /y update_new.py update.py
  echo Running updater again since it got updated.
  ..\python_embeded\python.exe .\update.py ..\ComfyUI\ --skip_self_update
)

cd ..
cd ComfyUI\custom_nodes

REM Clone ComfyUI-Manager
echo Installing ComfyUI-Manager...
git clone https://github.com/ltdrdata/ComfyUI-Manager.git
echo Installing missing nodes...
echo   -Impact pack
git clone https://github.com/ltdrdata/ComfyUI-Impact-Pack
cd ComfyUI-Impact-Pack
git clone https://github.com/ltdrdata/ComfyUI-Impact-Subpack impact_subpack
..\..\..\python_embeded\python.exe -s -m pip install -r requirements.txt --no-warn-script-location
..\..\..\python_embeded\python.exe -s -m pip install ultralytics --no-warn-script-location
cd ..

echo   -Controlnet aux
git clone https://github.com/Fannovel16/comfyui_controlnet_aux
cd comfyui_controlnet_aux
..\..\..\python_embeded\python.exe -s -m pip install -r requirements.txt --no-warn-script-location
cd ..

echo   -WAS suite aux
git clone https://github.com/WASasquatch/was-node-suite-comfyui
cd was-node-suite-comfyui
..\..\..\python_embeded\python.exe -s -m pip install -r requirements.txt --no-warn-script-location
cd ..

echo   -GGUF
git clone https://github.com/city96/ComfyUI-GGUF
cd ComfyUI-GGUF
..\..\..\python_embeded\python.exe -s -m pip install -r requirements.txt --no-warn-script-location
cd ..

echo   -Custom-Scripts
git clone https://github.com/pythongosssss/ComfyUI-Custom-Scripts

echo   -Ultimate SD Upscale
git clone https://github.com/ssitu/ComfyUI_UltimateSDUpscale --recursive

echo   -Various
git clone https://github.com/jamesWalker55/comfyui-various

echo   -rgthree
git clone https://github.com/rgthree/rgthree-comfy
cd rgthree-comfy
..\..\..\python_embeded\python.exe -s -m pip install -r requirements.txt --no-warn-script-location
cd ..

echo   -Florence2
git clone https://github.com/kijai/ComfyUI-Florence2
cd ComfyUI-Florence2
..\..\..\python_embeded\python.exe -s -m pip install -r requirements.txt --no-warn-script-location
cd ..

echo   -Inpaint-CropAndStitch
git clone https://github.com/lquesada/ComfyUI-Inpaint-CropAndStitch

echo   -Universal-Styler
git clone https://github.com/KoreTeknology/ComfyUI-Universal-Styler

echo   -Tensorops
git clone https://github.com/un-seen/comfyui-tensorops
cd comfyui-tensorops
..\..\..\python_embeded\python.exe -s -m pip install -r requirements.txt --no-warn-script-location
cd ..

echo   -X-Flux
git clone https://github.com/XLabs-AI/x-flux-comfyui
cd x-flux-comfyui
..\..\..\python_embeded\python.exe -s -m pip install -r requirements.txt --no-warn-script-location
cd ..

echo   -Efficiency Nodes
git clone https://github.com/jags111/efficiency-nodes-comfyui
cd efficiency-nodes-comfyui
..\..\..\python_embeded\python.exe -s -m pip install -r requirements.txt --no-warn-script-location
cd ..

echo   -Comfyroll Studio
git clone https://github.com/Suzie1/ComfyUI_Comfyroll_CustomNodes

echo   -art-venture
git clone https://github.com/sipherxyz/comfyui-art-venture
cd comfyui-art-venture
..\..\..\python_embeded\python.exe -s -m pip install -r requirements.txt --no-warn-script-location
cd ..

echo   -dream-project
git clone https://github.com/alt-key-project/comfyui-dream-project
cd comfyui-dream-project
..\..\..\python_embeded\python.exe -s -m pip install imageio pilgram scipy numpy==1.24.4 torchvision evalidate --no-warn-script-location
cd ..

echo   -KJNodes
git clone https://github.com/kijai/ComfyUI-KJNodes
cd ComfyUI-KJNodes
..\..\..\python_embeded\python.exe -s -m pip install -r requirements.txt --no-warn-script-location
cd ..

echo   -Lora-Auto-Trigger-Words
git clone https://github.com/idrirap/ComfyUI-Lora-Auto-Trigger-Words

echo   -Easy-Use
git clone https://github.com/yolain/ComfyUI-Easy-Use
cd ComfyUI-Easy-Use
..\..\..\python_embeded\python.exe -s -m pip install -r requirements.txt --no-warn-script-location
cd ..

echo   -mxToolkit
git clone https://github.com/Smirnov75/ComfyUI-mxToolkit

echo   -RvTools
git clone https://github.com/Rvage0815/ComfyUI-RvTools
cd ComfyUI-RvTools
..\..\..\python_embeded\python.exe -s -m pip install -r requirements.txt --no-warn-script-location
cd ..

echo   -Zssentials
git clone https://github.com/cubiq/ComfyUI_essentials
cd ComfyUI_essentials
..\..\..\python_embeded\python.exe -s -m pip install -r requirements.txt --no-warn-script-location

cd ..

echo   -PulID
git clone https://github.com/sipie800/ComfyUI-PuLID-Flux-Enhanced
cd ..\..\python_embeded
curl -L -o "insightface-0.7.3-cp311-cp311-win_amd64.whl" https://github.com/Gourieff/Assets/raw/main/Insightface/insightface-0.7.3-cp311-cp311-win_amd64.whl
.\python.exe -m pip install --use-pep517 facexlib
.\python.exe -m pip install git+https://github.com/rodjjo/filterpy.git
.\python.exe -m pip install onnxruntime==1.19.2 onnxruntime-gpu==1.15.1 insightface-0.7.3-cp311-cp311-win_amd64.whl
cd ..
cd ComfyUI\custom_nodes\ComfyUI-PuLID-Flux-Enhanced
..\..\..\python_embeded\python.exe -s -m pip install -r requirements.txt --no-warn-script-location
cd ..\..
cd models
git lfs install
git clone https://huggingface.co/Aitrepreneur/insightface 
mkdir pulid
cd pulid
curl -L -o "pulid_flux_v0.9.0.safetensors" https://huggingface.co/Aitrepreneur/FLX/resolve/main/pulid_flux_v0.9.0.safetensors?download=true
cd ..

REM Navigate to models folder
cd ..\models

REM Download VAE file
echo Downloading VAE file...
cd vae
curl -L -o ae.safetensors https://huggingface.co/UmeAiRT/FLUX-auto_installer/resolve/main/ae.safetensors?download=true

cd ..

REM Download CLIP files
echo Downloading CLIP files...
cd clip
curl -L -o clip_l.safetensors https://huggingface.co/UmeAiRT/FLUX-auto_installer/resolve/main/clip_l.safetensors?download=true
curl -L -o t5xxl_fp8_e4m3fn.safetensors https://huggingface.co/UmeAiRT/FLUX-auto_installer/resolve/main/t5xxl_fp8_e4m3fn.safetensors?download=true
curl -L -o t5xxl_fp16.safetensors https://huggingface.co/UmeAiRT/FLUX-auto_installer/resolve/main/t5xxl_fp16.safetensors?download=true
curl -L -o ViT-L-14-TEXT-detail-improved-hiT-GmP-TE-only-HF.safetensors https://huggingface.co/zer0int/CLIP-GmP-ViT-L-14/resolve/main/ViT-L-14-TEXT-detail-improved-hiT-GmP-TE-only-HF.safetensors?download=true

if "%DOWNLOAD_GGUF%"=="yes" (
    echo Downloading FLUX GGUF encoder Model...
    if /i "%FLUX_GGUF_CHOICE%"=="A" (
	curl -L -o t5-v1_1-xxl-encoder-Q8_0.gguf https://huggingface.co/city96/t5-v1_1-xxl-encoder-gguf/resolve/main/t5-v1_1-xxl-encoder-Q8_0.gguf?download=true
	) else if /i "%FLUX_GGUF_CHOICE%"=="B" (
	curl -L -o t5-v1_1-xxl-encoder-Q5_K_M.gguf https://huggingface.co/city96/t5-v1_1-xxl-encoder-gguf/resolve/main/t5-v1_1-xxl-encoder-Q5_K_M.gguf?download=true
	) else if /i "%FLUX_GGUF_CHOICE%"=="C" (
	curl -L -o t5-v1_1-xxl-encoder-Q3_K_L.gguf https://huggingface.co/city96/t5-v1_1-xxl-encoder-gguf/resolve/main/t5-v1_1-xxl-encoder-Q3_K_L.gguf?download=true
	) else if /i "%FLUX_GGUF_CHOICE%"=="D" (
	curl -L -o t5-v1_1-xxl-encoder-Q8_0.gguf https://huggingface.co/city96/t5-v1_1-xxl-encoder-gguf/resolve/main/t5-v1_1-xxl-encoder-Q8_0.gguf?download=true
	curl -L -o t5-v1_1-xxl-encoder-Q5_K_M.gguf https://huggingface.co/city96/t5-v1_1-xxl-encoder-gguf/resolve/main/t5-v1_1-xxl-encoder-Q5_K_M.gguf?download=true
	curl -L -o t5-v1_1-xxl-encoder-Q3_K_L.gguf https://huggingface.co/city96/t5-v1_1-xxl-encoder-gguf/resolve/main/t5-v1_1-xxl-encoder-Q3_K_L.gguf?download=true
	)
)
cd ..

REM Download UNET file based on user choice
echo Downloading UNET file...
cd unet
if "%INSTALL_TYPE%"=="fast-lowvram" (
    curl -L -o flux1-dev-fp8.safetensors https://huggingface.co/UmeAiRT/FLUX-auto_installer/resolve/main/flux1-dev-fp8.safetensors?download=true
) else (
    curl -L -o flux1-dev.sft https://huggingface.co/Aitrepreneur/FLX/resolve/main/flux1-dev.sft?download=true
)

if "%DOWNLOAD_GGUF%"=="yes" (
    echo Downloading FLUX GGUF Model...
    if /i "%FLUX_GGUF_CHOICE%"=="A" (
	curl -L -o flux1-dev-Q8_0.gguf https://huggingface.co/city96/FLUX.1-dev-gguf/resolve/main/flux1-dev-Q8_0.gguf?download=true
	) else if /i "%FLUX_GGUF_CHOICE%"=="B" (
	curl -L -o flux1-dev-Q5_K_S.gguf https://huggingface.co/city96/FLUX.1-dev-gguf/resolve/main/flux1-dev-Q5_K_S.gguf?download=true
	) else if /i "%FLUX_GGUF_CHOICE%"=="C" (
	curl -L -o flux1-dev-Q4_K_S.gguf https://huggingface.co/city96/FLUX.1-dev-gguf/resolve/main/flux1-dev-Q4_K_S.gguf?download=true
	) else if /i "%FLUX_GGUF_CHOICE%"=="D" (
	curl -L -o flux1-dev-Q8_0.gguf https://huggingface.co/city96/FLUX.1-dev-gguf/resolve/main/flux1-dev-Q8_0.gguf?download=true
	curl -L -o flux1-dev-Q5_K_S.gguf https://huggingface.co/city96/FLUX.1-dev-gguf/resolve/main/flux1-dev-Q5_K_S.gguf?download=true
	curl -L -o flux1-dev-Q4_K_S.gguf https://huggingface.co/city96/FLUX.1-dev-gguf/resolve/main/flux1-dev-Q4_K_S.gguf?download=true
	)
)

REM Download FLUX SCHNELL Model if user chose to
if "%DOWNLOAD_FLUX_SCHNELL%"=="yes" (
    echo Downloading FLUX SCHNELL Model...
    curl -L -o flux1-schnell-fp8.safetensors https://huggingface.co/UmeAiRT/FLUX-auto_installer/resolve/main/flux1-schnell-fp8.safetensors?download=true
)

cd ..

cd loras
REM Download FLUX SCHNELL Model if user chose to
if "%DOWNLOAD_FLUX_LORA%"=="yes" (
    echo Downloading UmeAiRT LoRAs...
    curl -L -o ume_sky_v2.safetensors https://huggingface.co/UmeAiRT/FLUX.1-dev-LoRA-Ume_Sky/resolve/main/ume_sky_v2.safetensors?download=true
    curl -L -o ume_modern_pixelart.safetensors https://huggingface.co/UmeAiRT/FLUX.1-dev-LoRA-Modern_Pixel_art/resolve/main/ume_modern_pixelart.safetensors?download=true
    curl -L -o ume_classic_Romanticism.safetensors https://huggingface.co/UmeAiRT/FLUX.1-dev-LoRA-Romanticism/resolve/main/ume_classic_Romanticism.safetensors?download=true
    curl -L -o ume_classic_impressionist.safetensors https://huggingface.co/UmeAiRT/FLUX.1-dev-LoRA-Impressionism/resolve/main/ume_classic_impressionist.safetensors?download=true
    curl -L -o ume_the-little-newspaper.safetensors https://huggingface.co/UmeAiRT/FLUX.1-dev-LoRA-Ume_J1900/resolve/main/umej1900.safetensors?download=true
    curl -L -o ume_knight.safetensors https://huggingface.co/UmeAiRT/FLUX.1-dev-LoRA-Ume_Knight/resolve/main/ume_gachaak.safetensors?download=true
)

cd ..

REM Download upscale model
echo Downloading upscale models...
cd upscale_models
curl -L -o 4x_NMKD-Siax_200k.pth https://huggingface.co/uwg/upscaler/resolve/main/ESRGAN/4x_NMKD-Siax_200k.pth?download=true
curl -L -o 4x-ClearRealityV1.pth https://huggingface.co/skbhadra/ClearRealityV1/resolve/main/4x-ClearRealityV1.pth?download=true

cd ..

mkdir LLM
mkdir .\xlabs\controlnets
cd xlabs\controlnets
REM Download FLUX ControlNet Models if user chose to
if "%DOWNLOAD_FLUX_CONTROLNET%"=="yes" (
    echo Downloading FLUX ControlNet Models...
    curl -L -o flux-canny-controlnet-v3.safetensors https://huggingface.co/XLabs-AI/flux-controlnet-canny-v3/resolve/main/flux-canny-controlnet-v3.safetensors?download=true
    curl -L -o flux-depth-controlnet-v3.safetensors https://huggingface.co/XLabs-AI/flux-controlnet-depth-v3/resolve/main/flux-depth-controlnet-v3.safetensors?download=true
    curl -L -o flux-hed-controlnet-v3.safetensors https://huggingface.co/XLabs-AI/flux-controlnet-hed-v3/resolve/main/flux-hed-controlnet-v3.safetensors?download=true
)

cd ..\..\..
mkdir .\user\default
echo Downloading comfy settings...
cd user\default
curl -L -o comfy.settings.json https://huggingface.co/UmeAiRT/FLUX-auto_installer/resolve/main/comfy.settings.json?download=true
echo Downloading comfy workflow...
mkdir .\workflows
cd workflows
curl -L -o "FLUX - TXT to IMG.json" https://huggingface.co/UmeAiRT/FLUX-auto_installer/resolve/main/FLUX-TXTtoIMG.json?download=true
curl -L -o "FLUX - IMG to IMG.json" https://huggingface.co/UmeAiRT/FLUX-auto_installer/resolve/main/FLUX-IMGtoIMG.json?download=true
curl -L -o "FLUX - Inpainting.json" https://huggingface.co/UmeAiRT/FLUX-auto_installer/resolve/main/FLUX-Inpainting.json?download=true
curl -L -o "FLUX - Upscale.json" https://huggingface.co/UmeAiRT/FLUX-auto_installer/resolve/main/FLUX-Upscale.json?download=true
curl -L -o "FLUX - CTRLNet DEPTH.json" https://huggingface.co/UmeAiRT/FLUX-auto_installer/resolve/main/FLUX-CTRLNetDEPTH.json?download=true
curl -L -o "FLUX - CTRLNet CANNY.json" https://huggingface.co/UmeAiRT/FLUX-auto_installer/resolve/main/FLUX-CTRLNetCANNY.json?download=true
curl -L -o "FLUX - CTRLNet HED.json" https://huggingface.co/UmeAiRT/FLUX-auto_installer/resolve/main/FLUX-CTRLNetHED.json?download=true
curl -L -o "FLUX - LoRA tester.json" https://huggingface.co/UmeAiRT/FLUX-auto_installer/resolve/main/FLUX-LoRATester.json?download=true
curl -L -o "UmeAiRT - Node check.json" https://huggingface.co/UmeAiRT/FLUX-auto_installer/resolve/main/UmeAiRT-Nodecheck.json?download=true
cd ..\..\..\..

cd ComfyUI
cd custom_nodes
mkdir ComfyUI-NAI-styler
cd ComfyUI-NAI-styler
mkdir CSV
cd CSV
curl -L -o "naifilters.csv" https://huggingface.co/UmeAiRT/FLUX-auto_installer/resolve/main/naifilters.csv?download=true
curl -L -o "naistyles.csv" https://huggingface.co/UmeAiRT/FLUX-auto_installer/resolve/main/naistyles.csv?download=true
curl -L -o "naitypes.csv" https://huggingface.co/UmeAiRT/FLUX-auto_installer/resolve/main/naitypes.csv?download=true

cd ..\..\..\..

REM Final steps based on user choice
if "%INSTALL_TYPE%"=="fast-lowvram" (
    echo Downloading special run file for fast-lowvram...
    curl -L -o "run_nvidia_gpu-LOWVRAM.bat" "https://huggingface.co/UmeAiRT/FLUX-auto_installer/resolve/main/run_nvidia_gpu-LOWVRAM.bat?download=true"
    echo ComfyUI and FLUX installed. Running ComfyUI...
    call "run_nvidia_gpu-LOWVRAM.bat"
) else (
    echo ComfyUI and FLUX installed. HAVE FUN ;)
)