torch
pillow
torchvision
transformers