An attempt of minimal code recreation of diffusion models.
This code is currently tested on the MNIST dataset.
To train your own tiny diffuser simply run
python3 src/train.py
from inside the root directory of the project.
To test the model you trained simply run
python3 src/test.py --ckpt_path [MODEL_CHECKPOINT]