changing the requirements to a cpu torch version that works (huggingf…

…ace#23483)
sheonhan · Jun 1, 2023 · 57b5bba · 57b5bba
1 parent e5a7550
commit 57b5bba
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/examples/flax/vision/requirements.txt b/examples/flax/vision/requirements.txt
@@ -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
+torchvision==0.12.0+cpu