transformers
accelerate>=0.26.0
transformers[torch]
datasets
torch
scikit-learn