transformers
torch
einops
accelerate
gradio
sentencepiece