From 75803f241f13efd7316bcda62363e4254c4006dc Mon Sep 17 00:00:00 2001
From: Lorenzo Stella <stellalo@amazon.com>
Date: Thu, 6 Feb 2025 16:29:32 +0100
Subject: [PATCH] Relax torch compatibility to <3 (#277)

*Issue #, if available:* Solves #273

*Description of changes:*


By submitting this pull request, I confirm that you can use, modify,
copy, and redistribute this contribution, under the terms of your
choice.
---
 pyproject.toml | 2 +-
 1 file changed, 1 insertion(+), 1 deletion(-)

diff --git a/pyproject.toml b/pyproject.toml
index 0980084..4566e23 100644
--- a/pyproject.toml
+++ b/pyproject.toml
@@ -11,7 +11,7 @@ readme = "README.md"
 license = { file = "LICENSE" }
 requires-python = ">=3.9"
 dependencies = [
-    "torch>=2.0,<2.6",         # package was tested on 2.2
+    "torch>=2.0,<3",         # package was tested on 2.2
     "transformers>=4.30,<4.48",
     "accelerate>=0.32,<1",
 ]