Spaces:
Runtime error
Runtime error
cocktailpeanut
commited on
Commit
•
5f128b1
1
Parent(s):
63465b9
update
Browse files- requirements.txt +1 -1
requirements.txt
CHANGED
@@ -5,7 +5,7 @@ huggingface-hub
|
|
5 |
#-f https://storage.googleapis.com/jax-releases/jax_cuda_releases.html
|
6 |
#jax[cuda12_pip]~=0.4.25
|
7 |
jax[cuda12]
|
8 |
-
jaxlib
|
9 |
ml_collections
|
10 |
numpy
|
11 |
orbax-checkpoint
|
|
|
5 |
#-f https://storage.googleapis.com/jax-releases/jax_cuda_releases.html
|
6 |
#jax[cuda12_pip]~=0.4.25
|
7 |
jax[cuda12]
|
8 |
+
jaxlib==0.4.21
|
9 |
ml_collections
|
10 |
numpy
|
11 |
orbax-checkpoint
|