Merge pull request #3 from getzep/local_embedding

use platform_machine for torch selection
getzep · Jun 15, 2023 · 26574a0 · 26574a0
2 parents d5703c3 + 9308881
commit 26574a0
Show file tree

Hide file tree

Showing 2 changed files with 3 additions and 3 deletions.
diff --git a/poetry.lock b/poetry.lock
diff --git a/pyproject.toml b/pyproject.toml
@@ -26,12 +26,12 @@ python-dotenv = "^1.0.0"
 
 [[tool.poetry.dependencies.torch]]
 version = "^2.0.1"
-markers = "sys_platform == 'linux'"
+markers = "platform_machine == 'x86_64'"
 source = "torch"
 
 [[tool.poetry.dependencies.torch]]
 version = "^2.0.1"
-markers = "sys_platform == 'darwin'"
+markers = "platform_machine != 'x86_64'"
 source = "PyPI"
 
 [[tool.poetry.source]]