cocktailpeanut commited on
Commit
5f128b1
1 Parent(s): 63465b9
Files changed (1) hide show
  1. 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