Metadata-Version: 2.1
Name: diffusion-model-framework
Version: 0.1.0
Summary: 
Home-page: https://github.com/sachahu1/Diffusion-Model-Framework
Author: Sacha Hu
Author-email: sachahu@mindsparkai.com
Requires-Python: >=3.9,<3.12
Classifier: Programming Language :: Python :: 3
Classifier: Programming Language :: Python :: 3.9
Classifier: Programming Language :: Python :: 3.10
Classifier: Programming Language :: Python :: 3.11
Requires-Dist: numpy (>=1.26.4,<2.0.0)
Requires-Dist: pillow-simd (>=9.0.0.post1,<10.0.0)
Requires-Dist: tensorboard (>=2.17.0,<3.0.0)
Requires-Dist: torch (>=2.4.0,<3.0.0)
Requires-Dist: torchvision (>=0.19.0,<0.20.0)
Requires-Dist: tqdm (>=4.66.2,<5.0.0)
Project-URL: Documentation, https://www.sachahu.com/docs/diffusion-model-framework/latest/index.html
Project-URL: Repository, https://github.com/sachahu1/Diffusion-Model-Framework
Description-Content-Type: text/markdown

# Diffusion Model Playground

![GitHub Actions Workflow Status](https://img.shields.io/github/actions/workflow/status/sachahu1/Diffusion-Model-Framework/run-tests.yaml?branch=main&label=Tests)
![GitHub Release](https://img.shields.io/github/v/release/sachahu1/Diffusion-Model-Framework)
![PyPI - Python Version](https://img.shields.io/pypi/pyversions/diffusion-model-framework)
![GitHub Repo stars](https://img.shields.io/github/stars/sachahu1/Diffusion-Model-Framework)

This repo is a playground for implementing, training and visualising diffusion 
models.

![](https://github.com/sachahu1/Diffusion-Model-Framework/main/assets/sampling-small.gif)

# Getting Started
This tool uses the [poetry python manager](https://python-poetry.org/).
To install the dependencies, simply run `poetry install`.

## Examples
To get started with this codebase, head to the [examples](./examples) folder to
find a simple example to get started with training your own model along with
an example to easily run inference.

