Spaces:
Build error
Build error
Update requirements.txt
Browse files- requirements.txt +1 -1
requirements.txt
CHANGED
@@ -1,6 +1,6 @@
|
|
1 |
numpy>=1.16.4
|
2 |
#jax>=0.2.6
|
3 |
-
|
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
|