delete cpu dependecies

mikel-brostrom · May 12, 2024 · 91755b0 · 91755b0
1 parent 5e48c74
commit 91755b0
Show file tree

Hide file tree

Showing 2 changed files with 3 additions and 6 deletions.
diff --git a/poetry.lock b/poetry.lock
diff --git a/pyproject.toml b/pyproject.toml
@@ -42,17 +42,14 @@ ftfy = "^6.1.3"
 gitpython = "^3.1.42"
 torch = [
     {version = "^2.2.1", source="pypi", markers = "sys_platform == 'darwin'"},
-    {version = "^2.2.1", source="torch_cuda121", markers = "sys_platform == 'linux'"},
-    {version = "^2.2.1", source="torch_cuda121", markers = "sys_platform == 'win32'"},
+    {version = "^2.2.1", source="torch_cuda121", markers = "sys_platform == 'linux' or sys_platform == 'win32'"},
 ]
 torchvision = [
     {version = "^0.17.1", source="pypi", markers = "sys_platform == 'darwin'"},
-    {version = "^0.17.1", source="torch_cuda121", markers = "sys_platform == 'linux'"},
-    {version = "^0.17.1", source="torch_cuda121", markers = "sys_platform == 'win32'"},
+    {version = "^0.17.1", source="torch_cuda121", markers = "sys_platform == 'linux' or sys_platform == 'win32'"},
 ]
 
 
-
 [tool.poetry.group.test.dependencies]
 pytest = "^8.0.2"
 isort = "^5.13.2"