gradio
torch
transformers
accelerate
einops
peft