cocktailpeanut commited on
Commit
4ee3632
1 Parent(s): d4f36e4
Files changed (2) hide show
  1. requirements-cpu.txt +1 -1
  2. requirements.txt +2 -1
requirements-cpu.txt CHANGED
@@ -2,7 +2,7 @@ einops
2
  flax
3
  gradio
4
  huggingface-hub
5
- #jax
6
  jaxlib
7
  ml_collections
8
  numpy
 
2
  flax
3
  gradio
4
  huggingface-hub
5
+ jax
6
  jaxlib
7
  ml_collections
8
  numpy
requirements.txt CHANGED
@@ -3,7 +3,8 @@ flax
3
  gradio
4
  huggingface-hub
5
  -f https://storage.googleapis.com/jax-releases/jax_cuda_releases.html
6
- jax[cuda12_pip]~=0.4.25
 
7
  jaxlib
8
  ml_collections
9
  numpy
 
3
  gradio
4
  huggingface-hub
5
  -f https://storage.googleapis.com/jax-releases/jax_cuda_releases.html
6
+ #jax[cuda12_pip]~=0.4.25
7
+ jax[cuda12_pip]==0.4.25
8
  jaxlib
9
  ml_collections
10
  numpy