← all repositories

yuanchenyang/smalldiffusion

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

smalldiffusion
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.

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