This is a simplified framework to train and run diffusion models.

To get started, simply install the python package from PyPI:

pip install diffusion-model-framework

Then, you're ready to start setting up your own training process. For help getting started with that, have a look at some of the examples.

Thanks to poetry, installing this package is very simple and can be done in a single command. Simply run:

poetry install

That's it, the package is installed. Move to the next section to learn how to use this package.

This tool uses poetry. If you already have poetry installed, please skip to the next section. Otherwise, let's first setup poetry.

To install poetry, simply run this command:

curl -sSL https://install.python-poetry.org | python3 -

You can find out more about poetry installation here.

That's it, poetry is set up.

Installing the package via poetry is very simple. Simply run:

poetry install

You can now start using the package to train your diffusion model. For example, try running:

poetry run python3 examples/train_model.py

Building a wheel is also possible via:

poetry build

First you will need to install the dependency group used for documentation.

poetry install --with documentation

Now you can build the documentation using Sphinx by running:

poetry run sphinx-build -M html docs/source/ docs/build