README.md

metadata

license: mit
tags:
  - 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