torch
torchvision
matplotlib
pillow
torchmetrics
numpy
huggingface_hub[cli]
transformers
accelerate
termcolor