File tree Expand file tree Collapse file tree 1 file changed +3
-18
lines changed
Expand file tree Collapse file tree 1 file changed +3
-18
lines changed Original file line number Diff line number Diff line change @@ -11,8 +11,6 @@ include = ["images/logo.png"]
1111
1212[tool .poetry .dependencies ]
1313python = " >=3.10,<3.14.0"
14-
15- # Core deps
1614rdflib = " 7.1.1"
1715networkx = " 3.2.1"
1816numpy = " *"
@@ -26,19 +24,11 @@ matplotlib = "*"
2624seaborn = " *"
2725scikit-learn = " *"
2826huggingface-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"
3229sentence-transformers = " ^5.1.0"
3330dspy = " ^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 ]
4434ruff = " *"
@@ -57,8 +47,3 @@ attr = "ontolearner.__version__"
5747[build-system ]
5848requires = [" poetry-core>=1.0.0" , " poetry-dynamic-versioning>=1.4.0" ]
5949build-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
You can’t perform that action at this time.
0 commit comments