gradio
torch
transformers
huggingface_hub
einops
accelerate