King-HAW / GMS

Official repository of Generative Medical Segmentation
https://arxiv.org/abs/2403.18198
48 stars 3 forks source link

The source of pre-trained SD VAE weight used in this work. #4

Open 16rq opened 3 months ago

16rq commented 3 months ago

Your work inspires me a lot! Although you provide detaild code and pretrained moel, I couldnt find the 'pretrained ckpt' source in its original work(stable diffusion). Could you provide it for more details about this pretrained VAE?

Thanks you!

King-HAW commented 3 months ago

Hi, thanks for your interest in our work. The SD VAE weight which could be downloaded from this link is extracted from the Stable Diffusion 2.1 (this is the original link). However, we only use the first stage VAE part, not the entire model.

16rq commented 3 months ago

okay, i got it. Thank you!