Skip to content

Commit a723d4f

Browse files
committed
🐛 revert back changes
1 parent d4dd11b commit a723d4f

File tree

1 file changed

+3
-18
lines changed

1 file changed

+3
-18
lines changed

pyproject.toml

Lines changed: 3 additions & 18 deletions
Original file line numberDiff line numberDiff line change
@@ -11,8 +11,6 @@ include = ["images/logo.png"]
1111

1212
[tool.poetry.dependencies]
1313
python = ">=3.10,<3.14.0"
14-
15-
# Core deps
1614
rdflib = "7.1.1"
1715
networkx = "3.2.1"
1816
numpy = "*"
@@ -26,19 +24,11 @@ matplotlib = "*"
2624
seaborn = "*"
2725
scikit-learn = "*"
2826
huggingface-hub = "^0.34.4"
29-
# LLM stack (CPU by default)
30-
torch = { version = "^2.8.0", source = "pytorch-cpu" }
31-
transformers = { version = "^4.56.0", extras = ["torch"] }
27+
torch = "^2.8.0"
28+
transformers = "^4.56.0"
3229
sentence-transformers = "^5.1.0"
3330
dspy = "^2.6.14"
34-
35-
[tool.poetry.extras]
36-
# GPU-only extras
37-
gpu = [
38-
"torch",
39-
"transformer",
40-
"bitsandbytes"
41-
]
31+
bitsandbytes="^0.45.1"
4232

4333
[tool.poetry.dev-dependencies]
4434
ruff = "*"
@@ -57,8 +47,3 @@ attr = "ontolearner.__version__"
5747
[build-system]
5848
requires = ["poetry-core>=1.0.0", "poetry-dynamic-versioning>=1.4.0"]
5949
build-backend = "poetry_dynamic_versioning.backend"
60-
61-
[[tool.poetry.source]]
62-
name = "pytorch-cpu"
63-
url = "https://download.pytorch.org/whl/cpu"
64-
default = false

0 commit comments

Comments
 (0)