Spaces:
Build error
Build error
Revert "revert to cpu config"
Browse filesThis reverts commit b30f1b864bc39ddb16e9e2bad8ed575ac2c481ce.
- demo/src/config.py +0 -2
- requirements.txt +2 -2
demo/src/config.py
CHANGED
@@ -41,6 +41,4 @@ class NerfConfig:
|
|
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 = 2
|
|
|
41 |
FOCAL = 555.5555155968841
|
42 |
# reduce CHUNK if OOM
|
43 |
CHUNK = 4096
|
|
|
|
|
44 |
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
|