accelerate
datasets
evaluate
torch>=2
transformers>=4.36.2
tqdm
einops
sentencepiece
protobuf==3.19.6
scikit-learn
matplotlib
pandas
numpy
