pytorch diffusers unconditional-image-generation diffusion-model-class

这个模型用于生成蝴蝶图像

from diffusers import DDPMPipeline
pipeline = DDPMPipeline.from_pretrained('DeepBird/my_custom_ddpm_train')
image = pipeline().images[0]
image