From b0b3d6c62675c965be53600dd175a8c1d84fe1f9 Mon Sep 17 00:00:00 2001 From: Emmanuel Schmidbauer Date: Mon, 20 Mar 2023 10:41:57 -0400 Subject: [PATCH] fix setup.py so pip install does not fail --- setup.py | 9 ++++++--- 1 file changed, 6 insertions(+), 3 deletions(-) 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=[ -- 2.45.2