mirror of
https://github.com/lucidrains/DALLE2-pytorch.git
synced 2025-12-19 17:54:20 +01:00
5063d192b6
just take care of the logic for AdamW and transformers used namedtuples for clip adapter embedding outputs
45 lines
1.0 KiB
Python
45 lines
1.0 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.0.71',
|
|
license='MIT',
|
|
description = 'DALL-E 2',
|
|
author = 'Phil Wang',
|
|
author_email = 'lucidrains@gmail.com',
|
|
url = 'https://github.com/lucidrains/dalle2-pytorch',
|
|
keywords = [
|
|
'artificial intelligence',
|
|
'deep learning',
|
|
'text to image'
|
|
],
|
|
install_requires=[
|
|
'click',
|
|
'einops>=0.4',
|
|
'einops-exts>=0.0.3',
|
|
'kornia>=0.5.4',
|
|
'pillow',
|
|
'torch>=1.10',
|
|
'torchvision',
|
|
'tqdm',
|
|
'vector-quantize-pytorch',
|
|
'x-clip>=0.5.1',
|
|
'youtokentome'
|
|
],
|
|
classifiers=[
|
|
'Development Status :: 4 - Beta',
|
|
'Intended Audience :: Developers',
|
|
'Topic :: Scientific/Engineering :: Artificial Intelligence',
|
|
'License :: OSI Approved :: MIT License',
|
|
'Programming Language :: Python :: 3.6',
|
|
],
|
|
)
|