transformers
datasets
torch
accelerate>=1.1.0
