abetlen commited on
Commit
a9dabe2
1 Parent(s): 25a1778
Files changed (2) hide show
  1. requirements-cpu.txt +1 -0
  2. requirements.txt +1 -0
requirements-cpu.txt CHANGED
@@ -1,5 +1,6 @@
1
  einops
2
  flax
 
3
  gradio
4
  huggingface-hub
5
  ml_collections
 
1
  einops
2
  flax
3
+ jax
4
  gradio
5
  huggingface-hub
6
  ml_collections
requirements.txt CHANGED
@@ -1,5 +1,6 @@
1
  einops
2
  flax
 
3
  gradio
4
  huggingface-hub
5
  ml_collections
 
1
  einops
2
  flax
3
+ jax
4
  gradio
5
  huggingface-hub
6
  ml_collections