torch
numpy
Pillow
safetensors
