dome272/Diffusion-Models-pytorch
A minimal PyTorch implementation of DDPM diffusion models for unconditional and conditional image generation.

This repository provides a clean, educational implementation of diffusion models following the DDPM paper exactly. It includes two variants: unconditional and conditional diffusion model training, with the conditional version supporting Classifier-Free Guidance and Exponential Moving Average techniques. The implementation uses a UNet architecture for the denoising network and supports training on custom datasets as well as sampling/generated image production.