Mau124's picture
Update README.md
956916b verified
metadata
license: mit
tags:
  - pytorch
  - diffusers
  - unconditional-image-generation
  - diffusion-models-class

Model Card for Simple Difussion model using Hugging Face diffusers library

This model is a diffusion model for unconditional image generation of cartoon faces using the cartoon10k dataset. The code for this model can be found on github.

The pipeline for this model was created following Diffusion Models Class 🧨

Usage

from diffusers import DDPMPipeline

pipeline = DDPMPipeline.from_pretrained('Mau124/diffusion-model-cartoon10k-96')
image = pipeline().images[0]
image

Results of generating randomly 16 96x96 images:

image/png