← all repositories

cloneofsimo/minDiffusion

A minimal, self-contained PyTorch implementation of DDPM diffusion models for generating images.

1.2k stars Python Image · Video · AudioLearning
minDiffusion
Velocity · 7d
+0.8
★ / day
Trend
steady
star history

This educational repository provides a clean implementation of Denoising Diffusion Probabilistic Models (DDPM) using PyTorch. It includes training scripts for MNIST, CIFAR-10, and CelebA datasets, with a simple UNet architecture and time embeddings. Users can run the self-contained superminddpm.py script directly or train on larger datasets with train_cifar10.py. The implementation generates images by iteratively denoising random samples, following the forward/backward diffusion process.

heatdrop uses Google Analytics to see which pages get read — nothing else. Your call. How we handle data.