Two crucial papers we have used in this project are https://arxiv.org/abs/2102.09672 and https://arxiv.org/abs/2006.11239
In this project we have used some code directly from https://github.com/openai/improved-diffusion
Activate your python environment and use poetry install to install the packages.
Find the poetry here: https://python-poetry.org/docs/
Remember to use poe precommit before the last commit.