gradio==4.26.0
torch==2.2.2
transformers==4.39.3
einops==0.7.0