From dda425dc6112b18c30b741765c585065249a4420 Mon Sep 17 00:00:00 2001 From: Phil Wang Date: Mon, 31 Aug 2020 12:45:27 -0700 Subject: [PATCH] add scaffold --- denoising_diffusion_pytorch/__init__.py | 1 + .../denoising_diffusion_pytorch.py | 11 ++++++++ setup.py | 27 +++++++++++++++++++ 3 files changed, 39 insertions(+) create mode 100644 denoising_diffusion_pytorch/__init__.py create mode 100644 denoising_diffusion_pytorch/denoising_diffusion_pytorch.py create mode 100644 setup.py diff --git a/denoising_diffusion_pytorch/__init__.py b/denoising_diffusion_pytorch/__init__.py new file mode 100644 index 0000000..97f6d91 --- /dev/null +++ b/denoising_diffusion_pytorch/__init__.py @@ -0,0 +1 @@ +from denoising_diffusion_pytorch.denoising_diffusion_pytorch import DenoisingDiffusion diff --git a/denoising_diffusion_pytorch/denoising_diffusion_pytorch.py b/denoising_diffusion_pytorch/denoising_diffusion_pytorch.py new file mode 100644 index 0000000..60558a2 --- /dev/null +++ b/denoising_diffusion_pytorch/denoising_diffusion_pytorch.py @@ -0,0 +1,11 @@ +import torch +from torch import nn, einsum +import torch.nn.functional as F +from einops import rearrange + +class DenoisingDiffusion(nn.Module): + def __init__(self): + super().__init__() + + def forward(self, x): + return x \ No newline at end of file diff --git a/setup.py b/setup.py new file mode 100644 index 0000000..2300d1d --- /dev/null +++ b/setup.py @@ -0,0 +1,27 @@ +from setuptools import setup, find_packages + +setup( + name = 'denoising-diffusion-pytorch', + packages = find_packages(), + version = '0.0.1', + license='MIT', + description = 'Denoising Diffusion Probabilistic Models - Pytorch', + author = 'Phil Wang', + author_email = 'lucidrains@gmail.com', + url = 'https://github.com/lucidrains/denoising-diffusion-pytorch', + keywords = [ + 'artificial intelligence', + 'generative models' + ], + install_requires=[ + 'torch', + 'einops' + ], + classifiers=[ + 'Development Status :: 4 - Beta', + 'Intended Audience :: Developers', + 'Topic :: Scientific/Engineering :: Artificial Intelligence', + 'License :: OSI Approved :: MIT License', + 'Programming Language :: Python :: 3.6', + ], +) \ No newline at end of file