torch
torchvision
omegaconf
einops
pytorch-lightning==1.9.5
taming-transformers
