Skip to content

GitLab

Unverified Commit 56d6d21b authored by Aryan's avatar Aryan Committed by GitHub
Browse files

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

* pin max torch version

* update

* Update setup.py
parent 89565e91
Hide whitespace changes
Inline Side-by-side
Showing with 1 addition and 1 deletion
setup.py
View file @ 56d6d21b
...@@ -130,7 +130,7 @@ _deps = [ ...@@ -130,7 +130,7 @@ _deps = [
"regex!=2019.12.17", "regex!=2019.12.17",
"requests", "requests",
"tensorboard", "tensorboard",
"torch>=1.4", "torch>=1.4,<2.5.0",
"torchvision", "torchvision",
"transformers>=4.41.2", "transformers>=4.41.2",
"urllib3<=2.0.0", "urllib3<=2.0.0",
......
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment