Hugging Face's logo

Spaces:
flax-community
/
DietNerf-Demo
Build error

App Files Community
alexlau commited on
Commit
b30f1b8
•
1 Parent(s): 39dcc54

revert to cpu config

Browse files
Files changed (2) hide show
  1. demo/src/config.py +3 -1
  2. requirements.txt +4 -4
demo/src/config.py CHANGED
@@ -41,4 +41,6 @@ class NerfConfig:
41
  FOCAL = 555.5555155968841
42
  # reduce CHUNK if OOM
43
  CHUNK = 4096
44
- DOWNSAMPLE = 2
 
 
 
41
  FOCAL = 555.5555155968841
42
  # reduce CHUNK if OOM
43
  CHUNK = 4096
44
+ # in CPU mode, suggest downsample=8
45
+ # in GPU mode, suggest downsample=2
46
+ DOWNSAMPLE = 8
requirements.txt CHANGED
@@ -1,8 +1,8 @@
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
 
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