torch
torchvision
timm
pillow
safetensors
transformers
einops
pydantic==2.10.6