hassiahk commited on
Commit
f31bc63
1 Parent(s): dda664b

Update requirements.txt

Browse files
Files changed (1) hide show
  1. requirements.txt +1 -1
requirements.txt CHANGED
@@ -1,6 +1,6 @@
1
  numpy>=1.16.4
2
  #jax>=0.2.6
3
- --upgrade jax[cuda111]
4
  jaxlib -f https://storage.googleapis.com/jax-releases/cuda111/jaxlib-0.1.67+cuda111-cp38-none-manylinux2010_x86_64.whl
5
  flax>=0.2.2
6
  #opencv-python>=4.4.0
 
1
  numpy>=1.16.4
2
  #jax>=0.2.6
3
+ jax[cuda111]
4
  jaxlib -f https://storage.googleapis.com/jax-releases/cuda111/jaxlib-0.1.67+cuda111-cp38-none-manylinux2010_x86_64.whl
5
  flax>=0.2.2
6
  #opencv-python>=4.4.0