From 56d6d21baeb203a249b3049835916bfb0bfad061 Mon Sep 17 00:00:00 2001
From: Aryan <aryan@huggingface.co>
Date: Sun, 20 Oct 2024 01:50:56 +0530
Subject: [PATCH] [CI] pin max torch version to fix CI errors (#9709)

* pin max torch version

* update

* Update setup.py
---
 setup.py | 2 +-
 1 file changed, 1 insertion(+), 1 deletion(-)

diff --git a/setup.py b/setup.py
index 89e18cab629b..7a8cc898e005 100644
--- a/setup.py
+++ b/setup.py
@@ -130,7 +130,7 @@
     "regex!=2019.12.17",
     "requests",
     "tensorboard",
-    "torch>=1.4",
+    "torch>=1.4,<2.5.0",
     "torchvision",
     "transformers>=4.41.2",
     "urllib3<=2.0.0",