[CI] pin max torch version to fix CI errors (#9709)

a-r-r-o-w · sayakpaul · commit 3a89338d49ae · 2024-12-23T13:02:17.000+05:30
* pin max torch version

* update

* Update setup.py
diff --git 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",
