Hugging Face's logo

Spaces:
flax-community
/
DietNerf-Demo
Runtime error

App Files Community
alexlau commited on
Commit
eaa31f6
•
1 Parent(s): bf2d3d5

revert to CPU and fix jax, jaxlib, flax ver

Browse files
Files changed (1) hide show
  1. requirements.txt +4 -4
requirements.txt CHANGED
@@ -1,9 +1,9 @@
1
  --find-links https://storage.googleapis.com/jax-releases/jax_releases.html
2
  jax[cuda111]
3
  numpy>=1.16.4
4
- # jax>=0.2.6
5
- # jaxlib>=0.1.57
6
- flax>=0.2.2
7
  # opencv-python>=4.4.0
8
  Pillow>=7.2.0
9
  pyyaml>=5.3.1
@@ -13,4 +13,4 @@ tensorflow-hub>=0.11.0
13
  transformers==4.8.2
14
  tqdm==4.61.2
15
  streamlit==0.84.1
16
- gdown==3.13.0
 
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.6
7
  # opencv-python>=4.4.0
8
  Pillow>=7.2.0
9
  pyyaml>=5.3.1
 
13
  transformers==4.8.2
14
  tqdm==4.61.2
15
  streamlit==0.84.1
16
+ gdown==3.13.0