omegaconf torch torchvision einops diffusers transformers pytorch_lightning==1.6.5 taming-transformers kornia scipy accelerate git+https://github.com/openai/CLIP.git@main#egg=clip git+https://github.com/davidbau/baukit.git