torchvision
safetensors
pyyaml
numpy
pillow