transformers
datasets
torch
accelerate