wip

mostly-ai · Jan 30, 2025 · 9712952 · 9712952
1 parent 1daea8c
commit 9712952
Show file tree

Hide file tree

Showing 2 changed files with 437 additions and 64 deletions.
diff --git a/pyproject.toml b/pyproject.toml
@@ -28,7 +28,6 @@ dependencies = [
     "setuptools>=75.0.0,<76",
     "numpy>=1.26.3,<2",
     "pandas~=2.2.0",
-    "torch>=2.5.1",
     "pyarrow>=16.0.0",
     "joblib>=1.3.0,<2",
     "psutil>=5.9.5,<6",  # upgrade when colab psutil is updated
@@ -41,6 +40,7 @@ dependencies = [
     "formatron>=0.4.11,<0.5",
     "json-repair>=0.30.0,<0.31",
     "rich>=13.9.4,<14",
+    "torch>=2.5.1",
 ]
 
 [project.optional-dependencies]
@@ -50,8 +50,7 @@ gpu = [
     "torch>=2.5.1",
 ]
 cpu = [
-    "torch==2.5.1+cpu; sys_platform != 'darwin'",
-    "torch==2.5.1; sys_platform == 'darwin'"
+    "torch>=2.5.1",
 ]
 
 [dependency-groups]
@@ -93,7 +92,7 @@ explicit = true
 
 [tool.uv.sources]
 torch = [
-  { index = "pytorch-cpu", extra = "cpu"},
+  { index = "pytorch-cpu", extra = "cpu", marker = "platform_system == 'linux'" },
 ]
 
 [tool.hatch.build.targets.sdist]