pip==23.1.2
torch
torchvision
numpy
tqdm
timm
pillow
diffusers
accelerate