Diffusion implemented with tensorflow 2

Usage

Install prerequisite packages

python3 -m pip install -r requirements.txt

Train unconditional diffusion on ImageNet

python3 train.py

Sampling image

python3 test.py --ckpt ckpt