torch==2.2.2
transformers==4.41.2
numpy==1.26.4
nltk==3.8.1
numerize==0.12
evaluate==0.4.1
absl-py==2.1.0
rouge-score==0.1.2
torchtyping==0.1.4
rich==13.7.1
accelerate==0.28.0
datasets==2.18.0
sentencepiece==0.2.0
protobuf==3.20.3
deepspeed==0.14.0
wandb==0.16.6
lightning==2.2.3
litdata==0.2.15
natsort==8.4.0
