mirror of
https://github.com/lucidrains/DALLE2-pytorch.git
synced 2025-12-19 09:44:19 +01:00
022c94e443
Added config files for training Changed example image generation to be more efficient Added configuration description to README Removed unused import
55 lines
1.3 KiB
Python
55 lines
1.3 KiB
Python
from setuptools import setup, find_packages
|
|
|
|
setup(
|
|
name = 'dalle2-pytorch',
|
|
packages = find_packages(exclude=[]),
|
|
include_package_data = True,
|
|
entry_points={
|
|
'console_scripts': [
|
|
'dalle2_pytorch = dalle2_pytorch.cli:main',
|
|
'dream = dalle2_pytorch.cli:dream'
|
|
],
|
|
},
|
|
version = '0.3.5',
|
|
license='MIT',
|
|
description = 'DALL-E 2',
|
|
author = 'Phil Wang',
|
|
author_email = 'lucidrains@gmail.com',
|
|
long_description_content_type = 'text/markdown',
|
|
url = 'https://github.com/lucidrains/dalle2-pytorch',
|
|
keywords = [
|
|
'artificial intelligence',
|
|
'deep learning',
|
|
'text to image'
|
|
],
|
|
install_requires=[
|
|
'click',
|
|
'clip-anytorch',
|
|
'coca-pytorch>=0.0.5',
|
|
'einops>=0.4',
|
|
'einops-exts>=0.0.3',
|
|
'embedding-reader',
|
|
'kornia>=0.5.4',
|
|
'numpy',
|
|
'pillow',
|
|
'resize-right>=0.0.2',
|
|
'rotary-embedding-torch',
|
|
'torch>=1.10',
|
|
'torchvision',
|
|
'tqdm',
|
|
'vector-quantize-pytorch',
|
|
'x-clip>=0.4.4',
|
|
'youtokentome',
|
|
'webdataset>=0.2.5',
|
|
'fsspec>=2022.1.0',
|
|
'torchmetrics[image]>=0.8.0'
|
|
],
|
|
classifiers=[
|
|
'Development Status :: 4 - Beta',
|
|
'Intended Audience :: Developers',
|
|
'Topic :: Scientific/Engineering :: Artificial Intelligence',
|
|
'License :: OSI Approved :: MIT License',
|
|
'Programming Language :: Python :: 3.6',
|
|
],
|
|
)
|