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