transformers==4.36.2
torch==2.2.2
gradio==4.12.0
accelerate==0.27.2
bitsandbytes==0.41.2
einops