fix: fix torch dependency for Intel Macs

Signed-off-by: Panos Vagenas <[email protected]>
DS4SD · Aug 28, 2024 · 8c83408 · 8c83408
1 parent 7a83e03
commit 8c83408
Show file tree

Hide file tree

Showing 2 changed files with 157 additions and 10 deletions.
diff --git a/poetry.lock b/poetry.lock
diff --git a/pyproject.toml b/pyproject.toml
@@ -22,8 +22,14 @@ packages = [
 
 [tool.poetry.dependencies]
 python = "^3.10"
-torch = "^2.2.2"
-torchvision = ">=0.17.2"
+torch = [
+  {markers = "sys_platform != 'darwin' or platform_machine != 'x86_64'", version = "^2.2.2"},
+  {markers = "sys_platform == 'darwin' and platform_machine == 'x86_64'", version = "~2.2.2"}
+]
+torchvision = [
+  {markers = "sys_platform != 'darwin' or platform_machine != 'x86_64'", version = "^0"},
+  {markers = "sys_platform == 'darwin' and platform_machine == 'x86_64'", version = "~0.17.2"}
+]
 onnxruntime = "^1.16.2"
 numpy = "^1.24.4"
 lxml = "^4.9.1"