diff --git a/setup.py b/setup.py index 08c69087..67e364d8 100644 --- a/setup.py +++ b/setup.py @@ -1,8 +1,11 @@ import setuptools -from pip.req import parse_requirements +try: # for pip >= 10 + from pip._internal.req import parse_requirements +except ImportError: # for pip <= 9.0.3 + from pip.req import parse_requirements -with open("README.old.md", "r", encoding="utf-8") as fh: +with open("README.md", "r", encoding="utf-8") as fh: long_description = fh.read() setuptools.setup( @@ -67,7 +70,7 @@ setuptools.setup( "g-mlp-pytorch", "x-clip", "x_transformers==1.0.4", - + "bitsandbytes==0.35.0", ], classifiers=[