transformers
gradio
einops
accelerate