bring in the simple tokenizer released by openai, but also plan on leaving room for custom tokenizer with yttm

This commit is contained in:
Phil Wang
2022-04-12 09:23:17 -07:00
parent 4ff6d021c9
commit 7cf1637d24
5 changed files with 262393 additions and 10 deletions

View File

@@ -3,6 +3,7 @@ from setuptools import setup, find_packages
setup(
name = 'dalle2-pytorch',
packages = find_packages(exclude=[]),
include_package_data = True,
version = '0.0.1',
license='MIT',
description = 'DALL-E 2',
@@ -16,8 +17,10 @@ setup(
],
install_requires=[
'einops>=0.4',
'einops-exts',
'torch>=1.6',
'x-clip>=0.4.1'
'x-clip>=0.4.1',
'yttm'
],
classifiers=[
'Development Status :: 4 - Beta',