torch>=2.2
torchvision>=0.17
pillow
huggingface_hub
termcolor
safetensors