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

* pin max torch version * update * Update setup.py
huggingface · Oct 19, 2024 · 56d6d21 · 56d6d21
1 parent 89565e9
commit 56d6d21
Showing 1 changed file with 1 addition and 1 deletion.
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",