gradio
transformers
torch
timm
einops
bitsandbytes
accelerate>=0.26.0