taming-transformers
omegaconf
einops
pytorch-lightning
lightning-fabric
open-clip-torch