torch>=1.7
transformers
diffusers
tqdm
numpy
matplotlib

[dev]
