gradio
torch
transformers
einops
accelerate