torch
transformers
accelerate
huggingface_hub
gradio
einops
einx
matplotlib
tqdm