Merge pull request #721 from allenai/hey-my-first-pr-to-olmo

Bump torch version
allenai · Oct 2, 2024 · 885bc22 · 885bc22
2 parents aa1863e + 2256ff5
commit 885bc22
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<2",
-    "torch>=2.1,<2.4",
+    "torch>=2.1,<2.5",
     "ai2-olmo-core==0.1.0",
     "omegaconf",
     "rich",