add scaffold

This commit is contained in:
Phil Wang
2020-08-31 12:45:27 -07:00
parent df94747c0c
commit dda425dc61
3 changed files with 39 additions and 0 deletions
+1
View File
@@ -0,0 +1 @@
from denoising_diffusion_pytorch.denoising_diffusion_pytorch import DenoisingDiffusion
@@ -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
+27
View File
@@ -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',
],
)