--extra-index-url https://download.pytorch.org/whl/cu121
torch==2.4.1
--extra-index-url https://download.pytorch.org/whl/cu121
torchvision==0.19.1
--extra-index-url https://download.pytorch.org/whl/cu121
torchaudio==2.4.1

transformers==4.48.0
tqdm==4.66.6
trl==0.8.3
openai
