This repo contains code for DDPM training. Based on Denoising Diffusion Probabilistic Models, Improved Denoising Diffusion Probabilistic Models and (in future) on Diffusion Models Beat GANs on Image Synthesis .
- Install dependencies with
pip install -r requirements.txt
. - To run training script with default configs, simply write
python diffusion_lightning
(default config: config/task/cifar10.yaml)
Command-line arguments provided with Hydra library.