phi2-grpo / requirements.txt
padmanabhbosamia's picture
Changes to bit and bytes
2e9b2d7 verified
raw
history blame contribute delete
227 Bytes
torch>=2.0.0
transformers>=4.36.0
datasets>=2.14.0
accelerate>=0.25.0
peft>=0.7.0
trl>=0.7.4
wandb>=0.16.0
gradio>=4.0.0
rich>=13.7.0
numpy>=1.24.0
tqdm>=4.66.0
scipy>=1.11.0
sentencepiece>=0.1.99
protobuf>=4.23.0