Upload AutoInstall Pytorch Triton Sage.bat

AutoInstall Pytorch Triton Sage.bat

@@ -0,0 +1,180 @@
+@echo off
+setlocal enabledelayedexpansion
+
+REM === SETUP PATHS ===
+set "ROOT=%CD%"
+set "PYTHON=%ROOT%\ComfyUI_windows_portable\python_embeded\python.exe"
+set "COMFY_DIR=%ROOT%\ComfyUI_windows_portable\ComfyUI"
+set "UPDATE_DIR=%ROOT%\ComfyUI_windows_portable\update"
+
+REM === CHECK PYTHON ===
+if not exist "%PYTHON%" (
+    echo [ERROR] Python not found at: %PYTHON%
+    pause
+    exit /b 1
+)
+
+REM === SHOW SYSTEM INFO ===
+echo.
+echo ======== ComfyUI Nightly Installer (Interactive) ========
+echo.
+
+REM Python version
+for /f "delims=" %%i in ('%PYTHON% -c "import platform; print(platform.python_version())"') do set PYVER=%%i
+echo [INFO] Python Version: %PYVER%
+
+REM Pip version
+for /f "delims=" %%i in ('%PYTHON% -m pip --version') do set PIPVER=%%i
+echo [INFO] Pip Version: %PIPVER%
+
+REM CPU Cores
+for /f %%i in ('powershell -NoProfile -Command "[Environment]::ProcessorCount"') do set "CPU_CORES=%%i"
+if not defined CPU_CORES set "CPU_CORES=16"
+echo [INFO] Detected CPU cores: %CPU_CORES%
+
+REM CUDA Version
+for /f "tokens=5 delims= " %%A in ('nvcc --version ^| findstr /C:"release"') do (
+    for /f "tokens=1 delims=," %%B in ("%%A") do set cuda_version=%%B
+)
+for /f "tokens=1 delims=." %%a in ("%cuda_version%") do set cuda_major=%%a
+for /f "tokens=2 delims=." %%b in ("%cuda_version%") do set cuda_minor=%%b
+set cuda_version=!cuda_major!.!cuda_minor!
+set "CLEAN_CUDA=%cuda_version:.=%"
+echo [INFO] Detected CUDA version: %cuda_version%
+
+echo.
+pause
+
+REM === PROMPTS ===
+
+REM Update pip and setuptools?
+set UPDATE_PIP=false
+echo.
+set /p "inp=Do you want to update pip and set setuptools to 70.2.0? (y/n): "
+if /i "!inp!"=="y" set UPDATE_PIP=true
+
+REM Install PyTorch?
+set INSTALL_TORCH=false
+echo.
+echo [1] Install PyTorch Nightly
+echo [2] Install PyTorch Stable
+echo [0] Skip PyTorch installation
+set /p "torch_choice=Choose PyTorch version: "
+if "!torch_choice!"=="1" set TORCH_NIGHTLY=true & set INSTALL_TORCH=true
+if "!torch_choice!"=="2" set TORCH_NIGHTLY=false & set INSTALL_TORCH=true
+
+REM Install Triton?
+set INSTALL_TRITON=false
+echo.
+echo [0] Skip Triton
+echo [1] Install Nightly Triton
+echo [2] Install Stable Triton
+echo [3] Install LeoMaxwell Triton
+set /p "triton_choice=Choose Triton version: "
+if not "!triton_choice!"=="0" set INSTALL_TRITON=true
+
+REM Install SageAttention?
+set INSTALL_SAGE=false
+echo.
+echo [1] Install SageAttention v1 (PyPI)
+echo [2] Install SageAttention v2 (Build from GitHub)
+echo [0] Skip SageAttention
+set /p "sage_choice=Choose SageAttention version: "
+if not "!sage_choice!"=="0" set INSTALL_SAGE=true
+
+REM Update ComfyUI?
+set UPDATE_COMFY=false
+echo.
+set /p "inp=Do you want to update ComfyUI via Git? (y/n): "
+if /i "!inp!"=="y" set UPDATE_COMFY=true
+
+REM Create Launcher?
+set MAKE_LAUNCHER=false
+echo.
+set /p "inp=Do you want to generate a launcher batch file for FP16+Sage? (y/n): "
+if /i "!inp!"=="y" set MAKE_LAUNCHER=true
+
+REM === RUN CHOSEN INSTALLS ===
+
+cd /d "%UPDATE_DIR%"
+
+if "%UPDATE_PIP%"=="true" (
+    echo [INFO] Updating pip and installing setuptools==70.2.0
+    %PYTHON% -m pip install --upgrade pip setuptools==70.2.0
+)
+
+if "%INSTALL_TORCH%"=="true" (
+    echo [INFO] Uninstalling old PyTorch versions
+    %PYTHON% -m pip uninstall -y torch torchvision torchaudio
+
+    if "%TORCH_NIGHTLY%"=="true" (
+        echo [INFO] Installing PyTorch Nightly for CUDA %CLEAN_CUDA%
+        %PYTHON% -m pip install --pre torch torchvision torchaudio --index-url https://download.pytorch.org/whl/nightly/cu%CLEAN_CUDA%
+    ) else (
+        echo [INFO] Installing PyTorch Stable for CUDA %CLEAN_CUDA%
+        %PYTHON% -m pip install torch torchvision torchaudio
+    )
+)
+
+if "%INSTALL_TRITON%"=="true" (
+    if "%triton_choice%"=="1" (
+        echo Installing Triton Nightly
+        %PYTHON% -m pip install -U --pre triton-windows
+    ) else if "%triton_choice%"=="2" (
+        echo Installing Triton Stable
+        %PYTHON% -m pip install triton-windows
+    ) else if "%triton_choice%"=="3" (
+        echo Installing LeoMaxwell Triton
+        %PYTHON% -m pip install https://github.com/leomaxwell973/Triton-3.3.0-UPDATE_FROM_3.2.0_and_FIXED-Windows-Nvidia-Prebuilt/releases/download/3.3.0_cu128_Py312/triton-3.3.0-cp312-cp312-win_amd64.whl
+    )
+)
+
+if "%INSTALL_SAGE%"=="true" (
+    if "%sage_choice%"=="1" (
+        echo Installing SageAttention v1
+        %PYTHON% -m pip install sageattention==1.0.6
+    ) else (
+        echo Installing SageAttention v2 from source
+        git clone https://github.com/thu-ml/SageAttention
+        cd SageAttention
+        set MAX_JOBS=%CPU_CORES%
+        %PYTHON% -m pip install .
+        cd ..
+        rmdir /s /q SageAttention
+    )
+)
+
+if "%UPDATE_COMFY%"=="true" (
+    echo Updating ComfyUI
+    pushd "%COMFY_DIR%"
+    git pull
+    popd
+)
+
+echo Installing ComfyUI Manager if not exists
+cd /d "%COMFY_DIR%\custom_nodes"
+if not exist ComfyUI-Manager (
+    git clone https://github.com/ltdrdata/ComfyUI-Manager.git
+)
+
+cd /d "%UPDATE_DIR%"
+%PYTHON% update.py ..\ComfyUI\
+if exist update_new.py (
+    move /y update_new.py update.py
+    %PYTHON% update.py ..\ComfyUI\ --skip_self_update
+)
+
+if "%MAKE_LAUNCHER%"=="true" (
+    echo Creating run_comfyui_fp16fast_sage.bat
+    set "LAUNCHER=%ROOT%\ComfyUI_windows_portable\run_comfyui_fp16fast_sage.bat"
+    (
+    echo call "C:\Program Files\Microsoft Visual Studio\2022\Community\VC\Auxiliary\Build\vcvars64.bat"
+    echo .\python_embeded\python.exe -s ComfyUI\main.py --use-sage-attention --windows-standalone-build --fast fp16_accumulation --listen 0.0.0.0
+    echo pause
+    ) > "%LAUNCHER%"
+)
+
+echo
+echo All selected components installed. You may now run ComfyUI.
+pause
+exit /b