Downloads · 30 days
328
1% of all-time downloads
ddpm-mnist
1aurent/ddpm-mnist
ddpm-mnist is a unconditional image generation model from 1aurent. Use it for the unconditional image generation task on the model card, and read the license before you ship it in a product. It is set up for diffusers. The card lists the license as mit.
All-time downloads
26.1K
Public
Parameters
1.1M
10.9 MB on disk
Likes
7
Public
Repo makeup
Click a slice to open those files.
.safetensors4.3 MB · 65%
At a glance
- Task
- Unconditional Image Generation
- Library
- diffusers
- License
- mit
- Access
- Public
- Created
- Oct 6, 2023
- Updated
- May 14, 2024
- SHA
- e6ee37a7
Datasets
Spaces
Model card
From the Hugging Face model README
Unconditional MNIST DDPM
Description
This model is a very lightweight UNet2D trained on the MNIST dataset.
This model is unconditional, meaning that you cannot pick which number you'd like to generate.
This model was trained in ~40min on an L4 GPU Google Colab instance. You can see the training logs in the Training metrics tab.
A conditional model is available at 1aurent/ddpm-mnist-conditional, though it is pretty buggy.
Usage
from diffusers import DDPMPipeline
pipeline = DDPMPipeline.from_pretrained('1aurent/ddpm-mnist')
image = pipeline().images[0]
image
- Task
- Unconditional Image Generation
- Library
- diffusers
- License
- mit
- Created
- Oct 6, 2023
- Updated
- May 14, 2024