keithtyser / diffusion-log-likelihood

Modifying stable diffusion to calculate negative log likelihood

Image captioning with text-to-image models

Team Members

Keith Tyser
Teona Bagashvili
Zihan Li

Need at least 24gb of VRAM and to add sd-v1-4.ckpt to your working directory. This can be downloaded from (https://huggingface.co/CompVis/stable-diffusion-v-1-4-original).