Commit
·
f02151e
1
Parent(s):
239ee43
Upload 2 files
Browse files- MANIFEST.in +1 -0
- 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 |
+
)
|