/diffusion_models

Minimal standalone example of diffusion model

Primary LanguageJupyter NotebookMIT LicenseMIT

Diffusion models

Open In Colab

A minimal standalone example of diffusion model. The objective is to understand the forward as well as reverse mapping process of diffusion models. The notebook contains the equations along side the code and some visualizations.

Hope you find it useful!

The forward trajectory:

Forward diffusion

Reverse trajectory:

Reverse diffusion