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

扩散模型生成蝴蝶图像

python

from diffusers import DDPMPipeline

pipeline = DDPMPipeline.from_pretrained('hairu00/ddpm-class-butterflies-00')

image = pipeline().images[0]

image