From d58faa5c38f88a1258e89e94f375bf46bce47769 Mon Sep 17 00:00:00 2001
From: Anh Uong <anh.uong@ibm.com>
Date: Fri, 18 Oct 2024 10:22:48 -0600
Subject: [PATCH] deps: torch<2.5 due to FA2 error with new version (#375)

Signed-off-by: Anh Uong <anh.uong@ibm.com>
---
 pyproject.toml | 2 +-
 1 file changed, 1 insertion(+), 1 deletion(-)

diff --git a/pyproject.toml b/pyproject.toml
index 021beeb57..909ea64d1 100644
--- a/pyproject.toml
+++ b/pyproject.toml
@@ -29,7 +29,7 @@ dependencies = [
 "numpy>=1.26.4,<2.0",
 "accelerate>=0.20.3,!=0.34,<1.1",
 "transformers>4.41,<4.50",
-"torch>=2.2.0,<3.0",
+"torch>=2.2.0,<2.5",
 "sentencepiece>=0.1.99,<0.3",
 "tokenizers>=0.13.3,<1.0",
 "tqdm>=4.66.2,<5.0",