einops
fire
huggingface_hub
matplotlib
numpy
opt-einsum
safetensors
torch
torchvision
tqdm
