Skip to content

Commit bcc052f

Browse files
(Deps) Require torch 2.x and minor updates (#1459)
1 parent 5892223 commit bcc052f

File tree

2 files changed

+15
-13
lines changed

2 files changed

+15
-13
lines changed

docs/source/installation.mdx

-1
Original file line numberDiff line numberDiff line change
@@ -128,7 +128,6 @@ Refer to the following table if you're using another CUDA Toolkit version.
128128

129129
```bash
130130
git clone https://github.com/bitsandbytes-foundation/bitsandbytes.git && cd bitsandbytes/
131-
pip install -r requirements-dev.txt
132131
cmake -DCOMPUTE_BACKEND=cuda -S .
133132
cmake --build . --config Release
134133
pip install -e . # `-e` for "editable" install, when developing BNB (otherwise leave that out)

pyproject.toml

+15-12
Original file line numberDiff line numberDiff line change
@@ -7,6 +7,10 @@ name = "bitsandbytes"
77
dynamic = ["version"]
88
description = "k-bit optimizers and matrix multiplication routines."
99
authors = [{name="Tim Dettmers", email="[email protected]"}]
10+
maintainers = [
11+
{name="Titus von Köller", email="[email protected]"},
12+
{name="Matthew Douglas", email="[email protected]"}
13+
]
1014
requires-python = ">=3.8"
1115
readme = "README.md"
1216
license = {file="LICENSE"}
@@ -26,7 +30,7 @@ classifiers = [
2630
"Intended Audience :: Developers",
2731
"Intended Audience :: Science/Research",
2832
"Operating System :: POSIX :: Linux",
29-
"Operating System :: MacOS",
33+
# "Operating System :: MacOS",
3034
"Operating System :: Microsoft :: Windows",
3135
"Programming Language :: C++",
3236
"Programming Language :: Python :: Implementation :: CPython",
@@ -38,10 +42,16 @@ classifiers = [
3842
"Topic :: Scientific/Engineering :: Artificial Intelligence"
3943
]
4044
dependencies = [
41-
"torch>=1.11,!=1.12.0",
45+
"torch~=2.0",
4246
"numpy>=1.17"
4347
]
4448

49+
[project.urls]
50+
homepage = "https://github.com/bitsandbytes-foundation/bitsandbytes"
51+
changelog = "https://github.com/bitsandbytes-foundation/bitsandbytes/blob/main/CHANGELOG.md"
52+
docs = "https://huggingface.co/docs/bitsandbytes/main"
53+
issues = "https://github.com/bitsandbytes-foundation/bitsandbytes/issues"
54+
4555
[project.optional-dependencies]
4656
benchmark = ["pandas", "matplotlib"]
4757
docs = ["hf-doc-builder==0.5.0"]
@@ -53,20 +63,13 @@ dev = [
5363
"wheel>=0.42,<1"
5464
]
5565
test = [
56-
"einops~=0.6.0",
57-
"lion-pytorch==0.0.6",
58-
"pytest~=7.4",
66+
"einops~=0.8.0",
67+
"lion-pytorch==0.2.3",
68+
"pytest~=8.3",
5969
"scipy>=1.10.1,<2; python_version < '3.9'",
6070
"scipy>=1.11.4,<2; python_version >= '3.9'",
6171
"transformers>=4.30.1,<5"
6272
]
63-
triton = ["triton~=2.0.0; sys_platform=='linux' and platform_machine=='x86_64'"]
64-
65-
[project.urls]
66-
homepage = "https://github.com/TimDettmers/bitsandbytes"
67-
changelog = "https://github.com/TimDettmers/bitsandbytes/blob/main/CHANGELOG.md"
68-
docs = "https://huggingface.co/docs/bitsandbytes/main"
69-
issues = "https://github.com/TimDettmers/bitsandbytes/issues"
7073

7174
[tool.setuptools]
7275
package-data = { "*" = ["libbitsandbytes*.*"] }

0 commit comments

Comments
 (0)