This is a clean implementation of DDPM

Installation

Create Anaconda environment

conda create -n ddpm_py311 python=3.11 --yes
conda activate ddpm_py311

Choose the CUDA version on the official PyTorch website: https://pytorch.org/

pip3 install torch torchvision torchaudio --index-url https://download.pytorch.org/whl/cu118

Install the remaining requirements

pip install -r requirements.txt

Train DDPM on MNIST

python main.py

Enjoy

python inference.py

More results