Skip to content

Commit 2dcd294

Browse files
committed
✨ update library dependencies
1 parent 832ae50 commit 2dcd294

File tree

1 file changed

+7
-7
lines changed

1 file changed

+7
-7
lines changed

pyproject.toml

Lines changed: 7 additions & 7 deletions
Original file line numberDiff line numberDiff line change
@@ -24,20 +24,20 @@ pathlib = "1.0.1"
2424
python-dotenv = "*"
2525
matplotlib = "*"
2626
seaborn = "*"
27-
scikit-learn = "1.6.1"
28-
huggingface-hub = "0.34.4"
29-
27+
scikit-learn = "*"
28+
huggingface-hub = "^0.34.4"
3029
# LLM stack (CPU by default)
3130
torch = { version = "^2.8.0", source = "pytorch-cpu" }
32-
transformers = "^4.56.0"
31+
transformers = { version = "^4.56.0", extras = ["torch"] }
3332
sentence-transformers = "^5.1.0"
3433
dspy = "^2.6.14"
3534

3635
[tool.poetry.extras]
3736
# GPU-only extras
3837
gpu = [
39-
"torch==^2.8.0", # GPU wheel (auto CUDA)
40-
"bitsandbytes=^0.45.1" # allow minor/patch updates
38+
"torch",
39+
"transformer",
40+
"bitsandbytes"
4141
]
4242

4343
[tool.poetry.dev-dependencies]
@@ -58,6 +58,6 @@ attr = "ontolearner.__version__"
5858
requires = ["poetry-core>=1.0.0", "poetry-dynamic-versioning>=1.4.0"]
5959
build-backend = "poetry_dynamic_versioning.backend"
6060

61-
[tool.poetry.source]
61+
[[tool.poetry.source]]
6262
name = "pytorch-cpu"
6363
url = "https://download.pytorch.org/whl/cpu"

0 commit comments

Comments
 (0)