torch
torchvision
transformers
huggingface_hub
accelerate
safetensors
Pillow
numpy
