Denoising Diffusion Probabilistic Models
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 .
Usage
- 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.