transformers==4.40.0
datasets==2.3.2
deepspeed==0.14.2
fairscale==0.4.5
ipython==8.14.0
nltk==3.8.1
tensorboard==2.16.2
tqdm==4.66.4
rouge_score==0.1.2
wandb==0.17.0
sentencepiece==0.2.0
ninja==1.11.1
accelerate==0.30.0
k_means_constrained==0.7.3
scikit-learn==1.4.2