yuanchenyang/smalldiffusion
A minimal diffusion model library providing training and sampling code for generative image models.

Velocity · 7d
+0.8
★ / day
Trend
→steady
star history
Smalldiffusion is a lightweight Python library focused on diffusion and flow models for generative tasks. It provides clean implementations of core model architectures (MLP, U-Net, DiT) along with training loops and sampling utilities. The library supports multiple prediction parameterizations (score, flow, data prediction) and includes example training scripts achieving competitive FID scores on FashionMNIST, CIFAR10, and ImageNet datasets.