alexlau commited on
Commit
40a686d
·
1 Parent(s): b30f1b8

revert to gpu mode

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