name of the pretrained UNET checkpoint of cin256-v2?
Closed this issue · 2 comments
bhosalems commented
https://ommer-lab.com/files/latent-diffusion/nitro/cin.zip I see that this path is being used in some notebooks online, although in the original latent diffusion github I could not find this path in download model scripts. I downloaded the weights but the default name of the file is model.ckpt, I just wanted to make sure if you just renamed this file to unet.ckpt?
bhosalems commented
I think the weight file was renamed because I can load the weights without any errors.
srikarym commented
Yeah this is the weights file of Imagenet LDM. We had to delete the class embedder weights from this ckpt, and change a couple of things in pytorch lightning's source code to allow fine-tuning