torch
torchvision
transformers<5
timm
einops
kornia
pillow
accelerate