Revert "revert to cpu config"

This reverts commit b30f1b864bc39ddb16e9e2bad8ed575ac2c481ce.

demo/src/config.py

@@ -41,6 +41,4 @@ class NerfConfig:
     FOCAL = 555.5555155968841
     # reduce CHUNK if OOM
     CHUNK = 4096
-    # in CPU mode, suggest downsample=8
-    # in GPU mode, suggest downsample=2
     DOWNSAMPLE = 2

requirements.txt

@@ -1,8 +1,8 @@
 --find-links https://storage.googleapis.com/jax-releases/jax_releases.html
 jax[cuda111]
 numpy>=1.16.4
-#jax>=0.2.6
-#jaxlib>=0.1.57
+# jax>=0.2.6
+# jaxlib>=0.1.57
 flax>=0.2.2
 # opencv-python>=4.4.0
 Pillow>=7.2.0