torch
torchvision
numpy
safetensors
pillow
tqdm