transformers
torch
torchvision
torchaudio
accelerate>=0.20.1