torch
torchvision
transformers
gradio
pillow
einops
timm
accelerate>=0.26.0
flash-attention