albumentations
matplotlib
diffusers==0.21.1
einops
huggingface_hub==0.25.1
natsort
pillow
PyYAML
regex
requests
timm
torch
torchvision
tqdm
transformers