numpy
accelerate>=0.20.3
transformers
torch
sentencepiece
tokenizers>=0.13.3
tqdm
trl
peft>=0.8.0
datasets>=2.15.0
fire

[aim]
aim==3.18.1

[dev]
wheel
packaging
ninja

[flash-attn]
flash-attn
