einops
numpy
pillow
torch>=1.6
torchvision
lpips
kornia
accelerate
tqdm
pycocotools
datasets
sentencepiece
open_clip_torch
transformers
