lyangas commited on
Commit
f08a300
·
1 Parent(s): d76bd51

Update pyproject.toml

Browse files

change source for torch-cpu

Files changed (1) hide show
  1. 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