torch
torchvision
transformers
numpy
accelerate
