-
Notifications
You must be signed in to change notification settings - Fork 0
Expand file tree
/
Copy pathpyproject.toml
More file actions
46 lines (40 loc) · 1.24 KB
/
pyproject.toml
File metadata and controls
46 lines (40 loc) · 1.24 KB
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
[tool.poetry]
name = "deductive_reasoning_framework"
version = "0.1.0"
description = "A Deductive Reasoning Framework using various ML models and RDF."
authors = ["David Wynter <daed@yambina.com>"]
[tool.poetry.dependencies]
python = ">=3.11,<3.13"
numpy = "^1.0.0"
streamlit = "^1.37.1"
rdflib = "^6.0.0"
bcrypt = "^3.2.0"
ray = {extras = ["tune"], version = "^2.0.0"}
requests = "^2.26.0"
pgmpy = "^0.1.25"
pymc = "5.16.2"
pyro-ppl = "^1.9.1"
owlready2 = "^0.46"
python-dotenv = "*"
llama-cpp-python = {version = "^0.2.73", optional=true} # Intel GPU support
huggingface-hub = "^0.22.2"
ctransformers = "^0.2.27"
# Intel oneAPI-optimized dependencies
torch = {version = ">=2.1.0", extras = ["cpu"]}
intel-extension-for-pytorch = {version = "^2.1.10", extras = ["xpu"]}
transformers = ">=4.30.0"
accelerate = ">=0.20.0"
intel-extension-for-transformers = "^1.3.1"
sentencepiece = ">=0.1.99"
mkl-devel = "^2024.1.0"
dpcpp-cpp-rt = "^2024.1.0"
# Optional but recommended for GPU acceleration
# sycl = "*"
# level-zero = "*"
[tool.poetry.extras]
xpu = ["llama-cpp-python"]
[tool.poetry.dev-dependencies]
pytest = "^6.2.5"
[build-system]
requires = ["poetry-core>=1.0.0"]
build-backend = "poetry.core.masonry.api"