datasets==2.18.0
pyarrow==15.0.2
tokenizers==0.15.2
torch==2.2.2
torchaudio==2.2.2
torchvision==0.17.2
transformers==4.39.3
triton==2.2.0
flash_attn==2.5.8
pyyaml
tensorboard
accelerate
