fix_compatibility.bat

@echo off
echo Fixing Python 3.12 compatibility issues for DPO training...
echo.

REM Uninstall problematic packages
echo Removing conflicting packages...
pip uninstall -y tensorflow keras protobuf

REM Install tf-keras for compatibility
echo Installing tf-keras...
pip install tf-keras

REM Install specific protobuf version
echo Installing compatible protobuf...
pip install protobuf==3.20.3

REM Install PyTorch with CUDA 11.8 support
echo Installing PyTorch...
pip install torch==2.1.2 torchvision==0.16.2 torchaudio==2.1.2 --index-url https://download.pytorch.org/whl/cu118

REM Install other dependencies with specific versions
echo Installing other dependencies...
pip install transformers==4.36.2
pip install accelerate==0.25.0
pip install peft==0.7.1
pip install trl==0.7.10
pip install bitsandbytes==0.42.0
pip install datasets
pip install pandas
pip install scipy
pip install sentencepiece

echo.
echo Done! Now try running: python train_dpo_hf_fixed.py
pause