transformers
datasets>
huggingface_hub>=0.25.2
torch
accelerate