Update pyproject.toml to allow torch 2.3

allenai · Jul 11, 2024 · a813afa · a813afa
1 parent b10ab4b
commit a813afa
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/pyproject.toml b/pyproject.toml
@@ -14,7 +14,7 @@ requires-python = ">=3.8"
 license = { file = "LICENSE" }
 dependencies = [
     "numpy",
-    "torch>=2.1,<2.3",
+    "torch>=2.1,<2.4",
     "ai2-olmo-core==0.1.0",
     "omegaconf",
     "rich",