taming-transformers
omegaconf
einops
pytorch-lightning>=2.2.1
open-clip-torch>=2.23.0
