@echo off
echo Installing Phi-4 Training Requirements
echo =====================================
echo.

REM Check if Python is available
where python >nul 2>&1
if %ERRORLEVEL% neq 0 (
    echo Python not found! Please make sure Python is installed and in your PATH.
    exit /b 1
)

echo Step 1: Installing base requirements...
python -m pip install -r requirements-base.txt
if %ERRORLEVEL% neq 0 (
    echo Failed to install base requirements.
    exit /b 1
)
echo Base requirements installed successfully.
echo.

echo Step 2: Installing additional requirements...
python -m pip install -r requirements.txt
if %ERRORLEVEL% neq 0 (
    echo Failed to install additional requirements.
    exit /b 1
)
echo Additional requirements installed successfully.
echo.

echo All required packages installed successfully!
echo To install optional flash-attention, run: python -m pip install -r requirements-flash.txt --no-build-isolation
echo.

pause