@echo off
SETLOCAL ENABLEDELAYEDEXPANSION



REM === Uninstall xformers (if present) ===
echo Uninstalling xformers if installed...
python.exe -m pip uninstall xformers -y >nul 2>&1

REM === Install required PyTorch packages ===
echo Installing torch, torchvision, and torchaudio...
python.exe -m pip install torch==2.7.0 torchvision torchaudio --index-url https://download.pytorch.org/whl/cu128
if errorlevel 1 (
    echo ERROR: Failed to install torch packages.
    pause
    exit /b 1
)

REM === Install xformers 0.0.30 ===
echo Installing xformers==0.0.30...
python.exe -m pip install xformers==0.0.30 --no-deps
if errorlevel 1 (
    echo ERROR: Failed to install xformers.
    pause
    exit /b 1
)


pause
ENDLOCAL