Update pyproject.toml
Browse fileschange source for torch-cpu
- pyproject.toml +10 -1
pyproject.toml
CHANGED
@@ -7,12 +7,21 @@ authors = ["lyangasovid <[email protected]>"]
|
|
7 |
[tool.poetry.dependencies]
|
8 |
python = "^3.8"
|
9 |
numpy = "1.22.4"
|
10 |
-
torch = "2.0.1"
|
11 |
scikit-learn = "1.3.0"
|
12 |
transformers = "4.29.2"
|
13 |
flask = "2.0.3"
|
14 |
xgboost = "1.7.6"
|
15 |
werkzeug = "2.2.2"
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
16 |
|
17 |
[tool.poetry.dev-dependencies]
|
18 |
|
|
|
7 |
[tool.poetry.dependencies]
|
8 |
python = "^3.8"
|
9 |
numpy = "1.22.4"
|
|
|
10 |
scikit-learn = "1.3.0"
|
11 |
transformers = "4.29.2"
|
12 |
flask = "2.0.3"
|
13 |
xgboost = "1.7.6"
|
14 |
werkzeug = "2.2.2"
|
15 |
+
torch = [
|
16 |
+
{version = "2.0.1", markers = "sys_platform == 'darwin'"},
|
17 |
+
{version = "2.0.1", markers = "sys_platform != 'darwin'", source = "torchcpu"}
|
18 |
+
]
|
19 |
+
|
20 |
+
[[tool.poetry.source]]
|
21 |
+
name = "torchcpu"
|
22 |
+
url = "https://download.pytorch.org/whl/cpu/"
|
23 |
+
default = false
|
24 |
+
secondary = true
|
25 |
|
26 |
[tool.poetry.dev-dependencies]
|
27 |
|