Skip to content

Commit 12153f8

Browse files
authored
fix: fix torch dependency for Intel Macs (#15)
Signed-off-by: Panos Vagenas <[email protected]>
1 parent 7a83e03 commit 12153f8

File tree

2 files changed

+157
-10
lines changed

2 files changed

+157
-10
lines changed

poetry.lock

Lines changed: 149 additions & 8 deletions
Some generated files are not rendered by default. Learn more about customizing how changed files appear on GitHub.

pyproject.toml

Lines changed: 8 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -22,8 +22,14 @@ packages = [
2222

2323
[tool.poetry.dependencies]
2424
python = "^3.10"
25-
torch = "^2.2.2"
26-
torchvision = ">=0.17.2"
25+
torch = [
26+
{markers = "sys_platform != 'darwin' or platform_machine != 'x86_64'", version = "^2.2.2"},
27+
{markers = "sys_platform == 'darwin' and platform_machine == 'x86_64'", version = "~2.2.2"}
28+
]
29+
torchvision = [
30+
{markers = "sys_platform != 'darwin' or platform_machine != 'x86_64'", version = "^0"},
31+
{markers = "sys_platform == 'darwin' and platform_machine == 'x86_64'", version = "~0.17.2"}
32+
]
2733
onnxruntime = "^1.16.2"
2834
numpy = "^1.24.4"
2935
lxml = "^4.9.1"

0 commit comments

Comments
 (0)