This repository provides a collection of TensorFlow 2 implementations/ ports for diffusion-based research.
The projects are self-contained and can be found here. All models are instances of tf.keras.Model, hence multi-gpu training can be enabled with minor adjustments thanks to tf.distribute.Strategy.
Under active development.
01/07/2024
- Initial release of this project.
- Released Iterative α-(de)Blending: a Minimalist Deterministic Diffusion Model by Heitz et al., in Special Interest Group on Graphics and Interactive Techniques (Siggraph), 2023.
- Released Inversion by Direct Iteration: An Alternative to Denoising Diffusion for Image Restoration by Delbracio & Milanfar, in Transactions on Machine Learning Research (TMLR), 2023.