transformers
torch
numpy
datasets
scikit-learn
accelerate>=0.26.0
pandas
tqdm