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 |
+
)
|