diff --git a/docs/source/installation.mdx b/docs/source/installation.mdx index 252575ce9..da4ab3b44 100644 --- a/docs/source/installation.mdx +++ b/docs/source/installation.mdx @@ -128,7 +128,6 @@ Refer to the following table if you're using another CUDA Toolkit version. ```bash git clone https://github.com/bitsandbytes-foundation/bitsandbytes.git && cd bitsandbytes/ -pip install -r requirements-dev.txt cmake -DCOMPUTE_BACKEND=cuda -S . cmake --build . --config Release pip install -e . # `-e` for "editable" install, when developing BNB (otherwise leave that out) diff --git a/pyproject.toml b/pyproject.toml index 3573b10f0..d861cc24f 100644 --- a/pyproject.toml +++ b/pyproject.toml @@ -7,6 +7,10 @@ name = "bitsandbytes" dynamic = ["version"] description = "k-bit optimizers and matrix multiplication routines." authors = [{name="Tim Dettmers", email="dettmers@cs.washington.edu"}] +maintainers = [ + {name="Titus von Köller", email="titus@huggingface.co"}, + {name="Matthew Douglas", email="matthew.douglas@huggingface.co"} +] requires-python = ">=3.8" readme = "README.md" license = {file="LICENSE"} @@ -26,7 +30,7 @@ classifiers = [ "Intended Audience :: Developers", "Intended Audience :: Science/Research", "Operating System :: POSIX :: Linux", - "Operating System :: MacOS", + # "Operating System :: MacOS", "Operating System :: Microsoft :: Windows", "Programming Language :: C++", "Programming Language :: Python :: Implementation :: CPython", @@ -38,10 +42,16 @@ classifiers = [ "Topic :: Scientific/Engineering :: Artificial Intelligence" ] dependencies = [ - "torch>=1.11,!=1.12.0", + "torch~=2.0", "numpy>=1.17" ] +[project.urls] +homepage = "https://github.com/bitsandbytes-foundation/bitsandbytes" +changelog = "https://github.com/bitsandbytes-foundation/bitsandbytes/blob/main/CHANGELOG.md" +docs = "https://huggingface.co/docs/bitsandbytes/main" +issues = "https://github.com/bitsandbytes-foundation/bitsandbytes/issues" + [project.optional-dependencies] benchmark = ["pandas", "matplotlib"] docs = ["hf-doc-builder==0.5.0"] @@ -53,20 +63,13 @@ dev = [ "wheel>=0.42,<1" ] test = [ - "einops~=0.6.0", - "lion-pytorch==0.0.6", - "pytest~=7.4", + "einops~=0.8.0", + "lion-pytorch==0.2.3", + "pytest~=8.3", "scipy>=1.10.1,<2; python_version < '3.9'", "scipy>=1.11.4,<2; python_version >= '3.9'", "transformers>=4.30.1,<5" ] -triton = ["triton~=2.0.0; sys_platform=='linux' and platform_machine=='x86_64'"] - -[project.urls] -homepage = "https://github.com/TimDettmers/bitsandbytes" -changelog = "https://github.com/TimDettmers/bitsandbytes/blob/main/CHANGELOG.md" -docs = "https://huggingface.co/docs/bitsandbytes/main" -issues = "https://github.com/TimDettmers/bitsandbytes/issues" [tool.setuptools] package-data = { "*" = ["libbitsandbytes*.*"] }