Upload AutoInstall Pytorch Triton Sage.bat

AutoInstall Pytorch Triton Sage.bat

@@ -112,7 +112,7 @@ if "%INSTALL_TORCH%"=="true" (
         %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
+        %PYTHON% -m pip install torch torchvision torchaudio --index-url https://download.pytorch.org/whl/cu%CLEAN_CUDA%
     )
 )
 
@@ -138,15 +138,18 @@ if "%INSTALL_SAGE%"=="true" (
         %PYTHON% -m pip install sageattention==1.0.6
     ) else (
         echo Installing SageAttention v2 from source
+        cd "%UPDATE_DIR%"
         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%"