wandb
datasets==2.2.1
transformers==4.19.2
pytorch-lightning==1.8.6
wget==3.2
rouge==1.0.0
tqdm==4.64.0
pandas==1.4.2
scikit-learn==1.1.0
higher==0.2.1
promptsource
deepspeed
sentencepiece
torch_kmeans
sentence-transformers
fsspec[adl]
click
rich