build: set cpu version for torch

twirapp · Feb 17, 2024 · 3e17198 · 3e17198
1 parent da58b63
commit 3e17198
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/pyproject.toml b/pyproject.toml
@@ -8,7 +8,7 @@ authors = [
 dependencies = [
     "uvicorn>=0.27.1",
     "litestar>=2.6.1",
-    "torch>=2.2.0",
+    "torch==2.2.0+cpu",
     "numpy>=1.26.4",
     "python-dotenv>=1.0.1",
 ]