transformers
sentencepiece
torch
accelerate
gradio==4.21.0