transformers
torch
pillow
gradio
kornia
torchvision
timm
psutil
einops