build(deps): revive bitsandbytes

kozistr · Jul 12, 2024 · fb493ae · fb493ae
1 parent 0821177
commit fb493ae
Show file tree

Hide file tree

Showing 2 changed files with 5 additions and 4 deletions.
diff --git a/poetry.lock b/poetry.lock
diff --git a/pyproject.toml b/pyproject.toml
@@ -47,6 +47,7 @@ classifiers = [
 python = ">=3.8,<4.0.0"
 numpy = { version = "*", python = ">=3.8" }
 torch = { version = ">=1.10", python = ">=3.8", source = "torch" }
+bitsandbytes = { version = "^0.43", optional = true }
 
 [tool.poetry.dev-dependencies]
 isort = { version = "^5", python = ">=3.8" }
@@ -55,6 +56,9 @@ ruff = "*"
 pytest = "*"
 pytest-cov = "*"
 
+[tool.poetry.extras]
+bitsandbytes = ["bitsandbytes"]
+
 [[tool.poetry.source]]
 name = "torch"
 url = "https://download.pytorch.org/whl/cpu"