transformers
torch
torchvision
accelerate