accelerate
numpy
transformers
matplotlib
diffusers
Pillow
torch
tqdm