Binxu Wang ([email protected])
Tutorial on Stable Diffusion Models at ML from Scratch seminar series at Harvard.
This tiny self-contained code base allows you to
- Rebuild the Stable Diffusion Model in a single Python script.
- Train your toy version of stable diffusion on classic datasets like MNIST, CelebA
- Playing with Stable Diffusion and inspecting the internal architecture of the models. Open in Colab
- Build your own Stable Diffusion UNet model from scratch in a notebook. (with < 300 lines of codes!) Open in Colab
- Build a Diffusion model (with UNet + cross attention) and train it to generate MNIST images based on the "text prompt". Open in Colab (exercise) Open in Colab (answer)
Music video generated from Stable Diffusion.