Spaces:
Build error
Build error
spec jax cpu install
Browse files- requirements.txt +1 -1
requirements.txt
CHANGED
@@ -1,7 +1,7 @@
|
|
1 |
#--find-links https://storage.googleapis.com/jax-releases/jax_releases.html
|
2 |
#jax[cuda111]
|
3 |
numpy>=1.16.4
|
4 |
-
jax==0.2.17
|
5 |
jaxlib==0.1.69
|
6 |
flax==0.3.4
|
7 |
# opencv-python>=4.4.0
|
|
|
1 |
#--find-links https://storage.googleapis.com/jax-releases/jax_releases.html
|
2 |
#jax[cuda111]
|
3 |
numpy>=1.16.4
|
4 |
+
jax[cpu]==0.2.17
|
5 |
jaxlib==0.1.69
|
6 |
flax==0.3.4
|
7 |
# opencv-python>=4.4.0
|