fix poetry install for torch

microsoft · Nov 5, 2024 · bdb8798 · bdb8798
1 parent b9e41ff
commit bdb8798
Show file tree

Hide file tree

Showing 2 changed files with 44 additions and 2 deletions.
diff --git a/poetry.lock b/poetry.lock
diff --git a/pyproject.toml b/pyproject.toml
@@ -51,7 +51,10 @@ nest-asyncio = "^1.6.0"
 altair-viewer = "^0.4.0"
 poetry = "^1.8.3"
 poethepoet = "^0.27.0"
-torch = {version = "^2.4.1", source = "pytorch_cpu"}
+torch = [
+    {version = "^2.5.1", source = "pytorch_cpu", markers = "platform_system != 'Darwin'"},
+    {version = "^2.5.1", markers = "platform_system == 'Darwin'"}
+]
 sentence-transformers = "^3.1.1"
 graspologic = "^3.4.1"
 future = "^1.0.0"