Merge pull request #652 from allenai/shanea/update-to-torch2.3

Update to torch 2.3
allenai · Jul 11, 2024 · d423c11 · d423c11
2 parents b10ab4b + a813afa
commit d423c11
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",