Upload AutoInstall Pytorch Triton Sage.bat
Browse files
AutoInstall Pytorch Triton Sage.bat
CHANGED
|
@@ -60,8 +60,13 @@ echo [1] Install PyTorch Nightly
|
|
| 60 |
echo [2] Install PyTorch Stable
|
| 61 |
echo [0] Skip PyTorch installation
|
| 62 |
set /p "torch_choice=Choose PyTorch version: "
|
| 63 |
-
if "
|
| 64 |
-
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 65 |
|
| 66 |
REM Install Triton?
|
| 67 |
set INSTALL_TRITON=false
|
|
@@ -180,7 +185,7 @@ if "%MAKE_LAUNCHER%"=="true" (
|
|
| 180 |
) > "%LAUNCHER%"
|
| 181 |
)
|
| 182 |
|
| 183 |
-
echo
|
| 184 |
echo All selected components installed. You may now run ComfyUI.
|
| 185 |
pause
|
| 186 |
exit /b
|
|
|
|
| 60 |
echo [2] Install PyTorch Stable
|
| 61 |
echo [0] Skip PyTorch installation
|
| 62 |
set /p "torch_choice=Choose PyTorch version: "
|
| 63 |
+
if "%torch_choice%"=="1" (
|
| 64 |
+
set TORCH_NIGHTLY=true
|
| 65 |
+
set INSTALL_TORCH=true
|
| 66 |
+
) else if "%torch_choice%"=="2" (
|
| 67 |
+
set TORCH_NIGHTLY=false
|
| 68 |
+
set INSTALL_TORCH=true
|
| 69 |
+
)
|
| 70 |
|
| 71 |
REM Install Triton?
|
| 72 |
set INSTALL_TRITON=false
|
|
|
|
| 185 |
) > "%LAUNCHER%"
|
| 186 |
)
|
| 187 |
|
| 188 |
+
echo.
|
| 189 |
echo All selected components installed. You may now run ComfyUI.
|
| 190 |
pause
|
| 191 |
exit /b
|