dvgodoy's picture
Upload README.md with huggingface_hub
e2eb13c
metadata
license: mit
tags:
  - pytorch
  - diffusers
  - unconditional-image-generation
  - diffusion-models-class

Diffusion Models 101

This model is a diffusion model for unconditional image generation of MNIST digits fine-tuned on Google's ddpm-cifar10-32 model

Usage

from diffusers import DDPMPipeline

pipeline = DDPMPipeline.from_pretrained('dvgodoy/ddpm-cifar10-32-mnist')
image = pipeline().images[0]
image