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

Usage

from diffusers import DDPMPipeline

pipeline = DDPMPipeline.from_pretrained('1aurent/ddpm-mnist')
image = pipeline().images[0]
image