sejamenath2023 commited on
Commit
f02151e
·
1 Parent(s): 239ee43

Upload 2 files

Browse files
Files changed (2) hide show
  1. MANIFEST.in +1 -0
  2. setup.py +55 -0
MANIFEST.in ADDED
@@ -0,0 +1 @@
 
 
1
+ recursive-include imagen_pytorch *.json
setup.py ADDED
@@ -0,0 +1,55 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ from setuptools import setup, find_packages
2
+ exec(open('imagen_pytorch/version.py').read())
3
+
4
+ setup(
5
+ name = 'imagen-pytorch',
6
+ packages = find_packages(exclude=[]),
7
+ include_package_data = True,
8
+ entry_points={
9
+ 'console_scripts': [
10
+ 'imagen_pytorch = imagen_pytorch.cli:main',
11
+ 'imagen = imagen_pytorch.cli:imagen'
12
+ ],
13
+ },
14
+ version = __version__,
15
+ license='MIT',
16
+ description = 'Imagen - unprecedented photorealism × deep level of language understanding',
17
+ author = 'Phil Wang',
18
+ author_email = '[email protected]',
19
+ long_description_content_type = 'text/markdown',
20
+ url = 'https://github.com/lucidrains/imagen-pytorch',
21
+ keywords = [
22
+ 'artificial intelligence',
23
+ 'deep learning',
24
+ 'transformers',
25
+ 'text-to-image',
26
+ 'denoising-diffusion'
27
+ ],
28
+ install_requires=[
29
+ 'accelerate>=0.23.0',
30
+ 'beartype',
31
+ 'click',
32
+ 'datasets',
33
+ 'einops>=0.7.0',
34
+ 'ema-pytorch>=0.0.3',
35
+ 'fsspec',
36
+ 'kornia',
37
+ 'numpy',
38
+ 'packaging',
39
+ 'pillow',
40
+ 'pydantic>=2',
41
+ 'pytorch-warmup',
42
+ 'sentencepiece',
43
+ 'torch>=1.6',
44
+ 'torchvision',
45
+ 'transformers',
46
+ 'tqdm'
47
+ ],
48
+ classifiers=[
49
+ 'Development Status :: 4 - Beta',
50
+ 'Intended Audience :: Developers',
51
+ 'Topic :: Scientific/Engineering :: Artificial Intelligence',
52
+ 'License :: OSI Approved :: MIT License',
53
+ 'Programming Language :: Python :: 3.6',
54
+ ],
55
+ )