File tree Expand file tree Collapse file tree 1 file changed +7
-7
lines changed
Expand file tree Collapse file tree 1 file changed +7
-7
lines changed Original file line number Diff line number Diff line change @@ -24,20 +24,20 @@ pathlib = "1.0.1"
2424python-dotenv = " *"
2525matplotlib = " *"
2626seaborn = " *"
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)
3130torch = { version = " ^2.8.0" , source = " pytorch-cpu" }
32- transformers = " ^4.56.0"
31+ transformers = { version = " ^4.56.0" , extras = [ " torch " ] }
3332sentence-transformers = " ^5.1.0"
3433dspy = " ^2.6.14"
3534
3635[tool .poetry .extras ]
3736# GPU-only extras
3837gpu = [
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__"
5858requires = [" poetry-core>=1.0.0" , " poetry-dynamic-versioning>=1.4.0" ]
5959build-backend = " poetry_dynamic_versioning.backend"
6060
61- [tool .poetry .source ]
61+ [[ tool .poetry .source ] ]
6262name = " pytorch-cpu"
6363url = " https://download.pytorch.org/whl/cpu"
You can’t perform that action at this time.
0 commit comments