Unverified Commit bcc052fd authored by Matthew Douglas's avatar Matthew Douglas Committed by GitHub
Browse files

(Deps) Require torch 2.x and minor updates (#1459)

parent 58922237
...@@ -128,7 +128,6 @@ Refer to the following table if you're using another CUDA Toolkit version. ...@@ -128,7 +128,6 @@ Refer to the following table if you're using another CUDA Toolkit version.
```bash ```bash
git clone https://github.com/bitsandbytes-foundation/bitsandbytes.git && cd bitsandbytes/ git clone https://github.com/bitsandbytes-foundation/bitsandbytes.git && cd bitsandbytes/
pip install -r requirements-dev.txt
cmake -DCOMPUTE_BACKEND=cuda -S . cmake -DCOMPUTE_BACKEND=cuda -S .
cmake --build . --config Release cmake --build . --config Release
pip install -e . # `-e` for "editable" install, when developing BNB (otherwise leave that out) pip install -e . # `-e` for "editable" install, when developing BNB (otherwise leave that out)
......
...@@ -7,6 +7,10 @@ name = "bitsandbytes" ...@@ -7,6 +7,10 @@ name = "bitsandbytes"
dynamic = ["version"] dynamic = ["version"]
description = "k-bit optimizers and matrix multiplication routines." description = "k-bit optimizers and matrix multiplication routines."
authors = [{name="Tim Dettmers", email="dettmers@cs.washington.edu"}] 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" requires-python = ">=3.8"
readme = "README.md" readme = "README.md"
license = {file="LICENSE"} license = {file="LICENSE"}
...@@ -26,7 +30,7 @@ classifiers = [ ...@@ -26,7 +30,7 @@ classifiers = [
"Intended Audience :: Developers", "Intended Audience :: Developers",
"Intended Audience :: Science/Research", "Intended Audience :: Science/Research",
"Operating System :: POSIX :: Linux", "Operating System :: POSIX :: Linux",
"Operating System :: MacOS", # "Operating System :: MacOS",
"Operating System :: Microsoft :: Windows", "Operating System :: Microsoft :: Windows",
"Programming Language :: C++", "Programming Language :: C++",
"Programming Language :: Python :: Implementation :: CPython", "Programming Language :: Python :: Implementation :: CPython",
...@@ -38,10 +42,16 @@ classifiers = [ ...@@ -38,10 +42,16 @@ classifiers = [
"Topic :: Scientific/Engineering :: Artificial Intelligence" "Topic :: Scientific/Engineering :: Artificial Intelligence"
] ]
dependencies = [ dependencies = [
"torch>=1.11,!=1.12.0", "torch~=2.0",
"numpy>=1.17" "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] [project.optional-dependencies]
benchmark = ["pandas", "matplotlib"] benchmark = ["pandas", "matplotlib"]
docs = ["hf-doc-builder==0.5.0"] docs = ["hf-doc-builder==0.5.0"]
...@@ -53,20 +63,13 @@ dev = [ ...@@ -53,20 +63,13 @@ dev = [
"wheel>=0.42,<1" "wheel>=0.42,<1"
] ]
test = [ test = [
"einops~=0.6.0", "einops~=0.8.0",
"lion-pytorch==0.0.6", "lion-pytorch==0.2.3",
"pytest~=7.4", "pytest~=8.3",
"scipy>=1.10.1,<2; python_version < '3.9'", "scipy>=1.10.1,<2; python_version < '3.9'",
"scipy>=1.11.4,<2; python_version >= '3.9'", "scipy>=1.11.4,<2; python_version >= '3.9'",
"transformers>=4.30.1,<5" "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] [tool.setuptools]
package-data = { "*" = ["libbitsandbytes*.*"] } package-data = { "*" = ["libbitsandbytes*.*"] }
......
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment