support bfloat16

2022-07-07 08:21:20 -04:00
parent 5f526e2109
commit da62298f06
9 changed files with 108 additions and 96 deletions
@@ -4,8 +4,8 @@ from pathlib import Path
 setuptools.setup(
    name='min-dalle',
    description = 'min(DALL·E)',
-    long_description=(Path(__file__).parent / "README.rst").read_text(),
-    version='0.2.29',
+    # long_description=(Path(__file__).parent / "README.rst").read_text(),
+    version='0.2.35',
    author='Brett Kuprel',
    author_email='brkuprel@gmail.com',
    url='https://github.com/kuprel/min-dalle',
@@ -15,8 +15,8 @@ setuptools.setup(
    ],
    license='MIT',
    install_requires=[
-        'torch>=1.10.0',
-        'typing_extensions>=4.1.0',
+        'torch>=1.11',
+        'typing_extensions>=4.1',
        'numpy>=1.21',
        'pillow>=7.1',
        'requests>=2.23'