BaconErie/Simple-Diffusion

Attempt at making a simple diffusion model from scratch in PyTorch.

Simple Diffusion

Attempt at making a simple diffusion model from scratch in PyTorch.

Almost everything works (forward noising, denoising model, reverse noising) except for actually generating coherent images.

My first experience with working with PyTorch, so although the diffusion model doesn't generate anything useful I learned how to work with PyTorch.

A quickly runnable notebook is availalbe on Kaggle: https://www.kaggle.com/code/baconerie/simple-diffusion

Sources

Gave helpful information about implementing a diffusion model

• Step by Step visual introduction to Diffusion Models by Kemal Erdem
• The Annotated Diffusion Model Niels Rogge and Kashif Rasul on HF