SUMIN/HuggingFace_transformer
1
0
Fork 0
Code Issues Pull Requests Actions 7 Packages Projects Releases Wiki Activity

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

Browse Source
This commit is contained in:
sshahrokhi
2023-05-22 09:58:55 -07:00
committed by GitHub
parent 5de2a6d5e5
commit 6f72e71f97
1 changed files with 2 additions and 2 deletions
Download Patch File Download Diff File Expand all files Collapse all files

4
examples/flax/vision/requirements.txt
View File

@@ -3,6 +3,6 @@ jaxlib>=0.1.59
flax>=0.3.5 flax>=0.3.5
optax>=0.0.8 optax>=0.0.8
-f https://download.pytorch.org/whl/torch_stable.html -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 -f https://download.pytorch.org/whl/torch_stable.html
torchvision==0.10.0+cpu torchvision==0.12.0+cpu