search

cvlab-stonybrook / PathLDM

Official Code for PathLDM: Text conditioned Latent Diffusion Model for Histopathology (WACV 2024)
33 stars 3 forks source link

name of the pretrained UNET checkpoint of cin256-v2? #5

Closed bhosalems closed 11 months ago

bhosalems commented 11 months ago

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 11 months ago

I think the weight file was renamed because I can load the weights without any errors.

srikarym commented 11 months ago

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