fix: correctly determine the OS platform to install torch

twirapp · May 17, 2024 · 0123387 · 0123387
1 parent 44612f5
commit 0123387
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/pyproject.toml b/pyproject.toml
@@ -11,8 +11,8 @@ dependencies = [
     "litestar>=2.6.1",
     "python-dotenv>=1.0.1",
     "numpy>=1.26.3",
-    "torch==2.2.0+cpu; sys_platform == 'aarch64' or sys_platform == 'x86_64'",
-    "torch==2.2.0; sys_platform != 'aarch64' and sys_platform != 'x86_64'",
+    "torch==2.2.0+cpu; platform_machine == 'x86_64'",
+    "torch==2.2.0; platform_machine != 'x86_64'",
 ]
 requires-python = ">=3.9"