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

* pin max torch version * update * Update setup.py

[CI] pin max torch version to fix CI errors (#9709)
* pin max torch version * update * Update setup.py
56d6d21b · Aryan · GitHub · 89565e91 · 56d6d21b
Unverified Commit 56d6d21b authored Oct 20, 2024 by Aryan Committed by GitHub Oct 20, 2024
Show whitespace changes
Inline Side-by-side

Showing with 1 addition and 1 deletion

setup.py setup.py +1 -1

No files found.
--- a/setup.py
+++ b/setup.py
@@ -130,7 +130,7 @@ _deps = [
    "regex!=2019.12.17",
    "requests",
    "tensorboard",
-    "torch>=1.4",
+    "torch>=1.4,<2.5.0",
    "torchvision",
    "transformers>=4.41.2",
    "urllib3<=2.0.0",