Skip to content

GitLab

"...git@developer.sourcefind.cn:chenpangpang/transformers.git" did not exist on "147fa37fb1fc19f8e77578a5958a1b0458205d5b"
Unverified Commit 6f72e71f authored by sshahrokhi's avatar sshahrokhi Committed by GitHub
Browse files

changing the requirements to a cpu torch version that works (#23483)

parent 5de2a6d5
Hide whitespace changes
Inline Side-by-side
Showing with 2 additions and 2 deletions
examples/flax/vision/requirements.txt
View file @ 6f72e71f
......@@ -3,6 +3,6 @@ jaxlib>=0.1.59
flax>=0.3.5
optax>=0.0.8
-f https://download.pytorch.org/whl/torch_stable.html
torch==1.9.0+cpu
torch==1.11.0+cpu
-f https://download.pytorch.org/whl/torch_stable.html
torchvision==0.10.0+cpu
\ No newline at end of file
torchvision==0.12.0+cpu
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