Skip to content

Commit 56d6d21

Browse files
authored
[CI] pin max torch version to fix CI errors (#9709)
* pin max torch version * update * Update setup.py
1 parent 89565e9 commit 56d6d21

File tree

1 file changed

+1
-1
lines changed

1 file changed

+1
-1
lines changed

setup.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -130,7 +130,7 @@
130130
"regex!=2019.12.17",
131131
"requests",
132132
"tensorboard",
133-
"torch>=1.4",
133+
"torch>=1.4,<2.5.0",
134134
"torchvision",
135135
"transformers>=4.41.2",
136136
"urllib3<=2.0.0",

0 commit comments

Comments
 (0)