From e18a29ed651f9b6ff45d5839c06457904f4e0e8f Mon Sep 17 00:00:00 2001
From: Costa Huang <costa.huang@outlook.com>
Date: Thu, 19 Sep 2024 12:57:43 -0400
Subject: [PATCH 1/2] Bump torch version

---
 pyproject.toml | 2 +-
 1 file changed, 1 insertion(+), 1 deletion(-)

diff --git a/pyproject.toml b/pyproject.toml
index ab034ff2f..542a2c107 100644
--- a/pyproject.toml
+++ b/pyproject.toml
@@ -14,7 +14,7 @@ requires-python = ">=3.8"
 license = { file = "LICENSE" }
 dependencies = [
     "numpy",
-    "torch>=2.1,<2.4",
+    "torch>=2.4",
     "ai2-olmo-core==0.1.0",
     "omegaconf",
     "rich",

From 7f57a4abf6bc5911614da6dfcf46086fc9eba458 Mon Sep 17 00:00:00 2001
From: Costa Huang <costa.huang@outlook.com>
Date: Thu, 19 Sep 2024 15:47:18 -0400
Subject: [PATCH 2/2] Update pyproject.toml

Co-authored-by: Shane A <shanea@allenai.org>
---
 pyproject.toml | 2 +-
 1 file changed, 1 insertion(+), 1 deletion(-)

diff --git a/pyproject.toml b/pyproject.toml
index 542a2c107..f114bd7d3 100644
--- a/pyproject.toml
+++ b/pyproject.toml
@@ -14,7 +14,7 @@ requires-python = ">=3.8"
 license = { file = "LICENSE" }
 dependencies = [
     "numpy",
-    "torch>=2.4",
+    "torch>=2.1,<2.5",
     "ai2-olmo-core==0.1.0",
     "omegaconf",
     "rich",