gradio
transformers
accelerate==0.29.3
torch