From fac4dc5130121383b985f1f40ec251db5e46b30f Mon Sep 17 00:00:00 2001
From: lewtun <lewis.c.tunstall@gmail.com>
Date: Wed, 22 Jan 2025 11:58:26 +0100
Subject: [PATCH] Relax upper bound on torch

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

diff --git a/pyproject.toml b/pyproject.toml
index f60e610ea..d20d9b68e 100644
--- a/pyproject.toml
+++ b/pyproject.toml
@@ -57,7 +57,7 @@ dependencies = [
     "transformers>=4.38.0",
     "accelerate",
     "huggingface_hub>=0.23.0",
-    "torch>=2.0,<2.5",
+    "torch>=2.0,<3.0",
     "GitPython>=3.1.41", # for logging
     "datasets>=2.14.0",
     "numpy<2",  # pinned to avoid incompatibilities