torch>=2.2.0
transformers==4.44.2
accelerate
gradio>=4.0.0
Pillow
sentencepiece
timm 
einops