Spaces:
Build error
Build error
revert to gpu mode
Browse files- demo/src/config.py +1 -1
- 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 =
|
|
|
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 |
-
|
2 |
-
|
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
|