diff options
Diffstat (limited to 'setup.py')
| -rw-r--r-- | setup.py | 22 |
1 files changed, 11 insertions, 11 deletions
@@ -6,27 +6,27 @@ import os from setuptools import setup, find_packages - def read(fname): return open(os.path.join(os.path.dirname(__file__), fname)).read() +version = os.getenv("CUDA_VERSION", "cpu") + setup( - name = f"bitsandbytes-cuda{os.environ['CUDA_VERSION']}", - version = "0.26.0", - author = "Tim Dettmers", - author_email = "dettmers@cs.washington.edu", - description = ("8-bit optimizers and quantization routines."), - license = "MIT", - keywords = "gpu optimizers optimization 8-bit quantization compression", - url = "http://packages.python.org/bitsandbytes", + name="bitsandbytes", + version=f"0.26.0+{version}", + author="Tim Dettmers", + author_email="dettmers@cs.washington.edu", + description="8-bit optimizers and quantization routines.", + license="MIT", + keywords="gpu optimizers optimization 8-bit quantization compression", + url="http://packages.python.org/bitsandbytes", packages=find_packages(), package_data={'': ['libbitsandbytes.so']}, long_description=read('README.md'), - long_description_content_type = 'text/markdown', + long_description_content_type='text/markdown', classifiers=[ "Development Status :: 4 - Beta", 'Topic :: Scientific/Engineering :: Artificial Intelligence' ], ) - |