Upload 2 files

UmeAiRT-AllinOne-Auto_install.bat

@@ -14,7 +14,7 @@ echo ---------------------------------------------------------------------------
 type banner.txt
 echo -------------------------------------------------------------------------------
 echo                     ComfyUI - WAN2.1 - All in one installer                          
-echo                                                                     V2.1          
+echo                                                                     V2.2          
 echo -------------------------------------------------------------------------------
 del /f banner.txt
 
@@ -55,7 +55,7 @@ if %errorlevel% NEQ 0 (
 
 REM Download ComfyUI
 echo Downloading ComfyUI...
-curl -L -o ComfyUI_windows_portable_nvidia.7z https://github.com/comfyanonymous/ComfyUI/releases/download/v0.3.29/ComfyUI_windows_portable_nvidia.7z
+curl -L -o ComfyUI_windows_portable_nvidia.7z https://github.com/comfyanonymous/ComfyUI/releases/download/v0.3.30/ComfyUI_windows_portable_nvidia.7z
 
 REM Extract ComfyUI
 echo Extracting ComfyUI...
@@ -201,6 +201,20 @@ REM Final steps based on user choice
 curl -L -o "%basePath%/run_nvidia_gpu-LOWVRAM.bat" https://huggingface.co/UmeAiRT/ComfyUI-Auto_installer/resolve/main/scripts/run_nvidia_gpu-LOWVRAM.bat?download=true
 curl -L -o "%basePath%/run_nvidia_gpu-sageattention.bat" https://huggingface.co/UmeAiRT/ComfyUI-Auto_installer/resolve/main/scripts/run_nvidia_gpu-sageattention.bat?download=true
 
+:CHOOSE_SAGEATTENTION
+REM Ask user for installation type
+echo Do you want to install SageAttention?
+set /p "MODELS=Enter your choice (Y or N) and press Enter: "
+
+if /i "%MODELS%"=="Y" (
+    call "%installPath%\UmeAiRT-WAN2.1-Sageattention_install.bat"
+) else if /i "%MODELS%"=="N" (
+    REM Do nothing, just continue
+) else (
+    echo Invalid choice. Please enter Y or N.
+    goto CHOOSE_SAGEATTENTION
+)
+
 :CHOOSE_DOWNLOAD_WAN
 REM Ask user for installation type
 echo Would you like to download WAN models?

UmeAiRT-WAN2.1-Sageattention_install.bat

@@ -68,6 +68,12 @@ winget install --id Microsoft.VisualStudio.2022.BuildTools -e --source winget --
 
 echo Installing Triton...
 "%pythonPath%\python.exe" -s -m pip install triton-windows==3.2.0.post17 >> "%basePath%\logs\SageAttention.txt" 2>&1
+curl -L -o "%basePath%\python_embeded\triton-3.2.0-cp310-cp310-win_amd64.whl" https://github.com/woct0rdho/triton-windows/releases/download/v3.2.0-windows.post10/triton-3.2.0-cp310-cp310-win_amd64.whl  >> "%installPath%\logs\SageAttention.txt" 2>&1
+curl -L -o "%basePath%\python_embeded\triton-3.2.0-cp311-cp311-win_amd64.whl" https://github.com/woct0rdho/triton-windows/releases/download/v3.2.0-windows.post10/triton-3.2.0-cp311-cp311-win_amd64.whl  >> "%installPath%\logs\SageAttention.txt" 2>&1
+curl -L -o "%basePath%\python_embeded\triton-3.2.0-cp312-cp312-win_amd64.whl" https://github.com/woct0rdho/triton-windows/releases/download/v3.2.0-windows.post10/triton-3.2.0-cp312-cp312-win_amd64.whl  >> "%installPath%\logs\SageAttention.txt" 2>&1
+"%basePath%\python_embeded\python.exe" -m pip install "%basePath%\python_embeded\triton-3.2.0-cp310-cp310-win_amd64.whl"  >> "%installPath%\logs\SageAttention.txt" 2>&1
+"%basePath%\python_embeded\python.exe" -m pip install "%basePath%\python_embeded\triton-3.2.0-cp311-cp311-win_amd64.whl"  >> "%installPath%\logs\SageAttention.txt" 2>&1
+"%basePath%\python_embeded\python.exe" -m pip install "%basePath%\python_embeded\triton-3.2.0-cp312-cp312-win_amd64.whl"  >> "%installPath%\logs\SageAttention.txt" 2>&1
 
 echo Downloading Python include/libs...
 curl -L -o "%pythonPath%\python_3.12.9_include_libs.zip" https://huggingface.co/UmeAiRT/ComfyUI-Auto_installer/resolve/main/others/python_3.12.9_include_libs.zip?download=true >> "%basePath%\logs\SageAttention.txt" 2>&1