search

showlab / DatasetDM

[NeurIPS2023] DatasetDM:Synthesizing Data with Perception Annotations Using Diffusion Models
https://weijiawu.github.io/DatasetDM_page/
290 stars 12 forks source link

About the weights of pre-trained diffusion model #14

Closed BlingHe closed 8 months ago

BlingHe commented 8 months ago

Hi,

Thanks for your great work! Could you provide some guidance about the weights of pre-trained diffusion models which you used for training and generating? In the code, the pathes of pre-trained models are defined by "./dataset/ckpts/imagenet/xxx". I am not very clear about how I can get these weights. And I also want to know why not use the weights of pre-trained stable diffusion models directly here.

image

Best, Jing

weijiawu commented 8 months ago

Thanks for your attention. As mentioned in the README (https://github.com/showlab/DatasetDM/blob/main/README.md), download the weights and configuration files of SD 1.4 (https://huggingface.co/CompVis/stable-diffusion-v1-4/tree/main) and place them in the ./dataset/ckpts directory.

BlingHe commented 8 months ago

Thanks for your attention. As mentioned in the README (https://github.com/showlab/DatasetDM/blob/main/README.md), download the weights and configuration files of SD 1.4 (https://huggingface.co/CompVis/stable-diffusion-v1-4/tree/main) and place them in the ./dataset/ckpts directory.

Thank you for your kind reply! I have solved this problem by modifying the code by:

image