requirements.txt

trl>=0.4.7
flash_attn==2.7.4.post1
transformers==4.48.2
accelerate==1.3.0
peft==0.13.2
torch==1.13.1+cu117  # CUDA version might vary depending on your system
datasets==2.10.1
trfl==0.5.6
tqdm==4.66.0
tokenizers==0.13.2