einops>=0.3
fire
numpy
pillow
pytorch-fid
retry
torch
torchvision
tqdm
