Spaces:
Sleeping
Sleeping
kyleleey
commited on
Commit
•
650d9ac
1
Parent(s):
e14627f
remove tcnn
Browse files- Dockerfile +0 -2
- video3d/render/mlptexture.py +3 -3
Dockerfile
CHANGED
@@ -57,8 +57,6 @@ RUN git clone -b v0.7.4 https://github.com/facebookresearch/pytorch3d.git /home/
|
|
57 |
WORKDIR /home/user/pytorch3d-0.7.4
|
58 |
RUN python setup.py install --user
|
59 |
|
60 |
-
RUN pip install git+https://github.com/NVlabs/tiny-cuda-nn/#subdirectory=bindings/torch
|
61 |
-
|
62 |
# Set the working directory to the user's home directory
|
63 |
WORKDIR $HOME/app
|
64 |
|
|
|
57 |
WORKDIR /home/user/pytorch3d-0.7.4
|
58 |
RUN python setup.py install --user
|
59 |
|
|
|
|
|
60 |
# Set the working directory to the user's home directory
|
61 |
WORKDIR $HOME/app
|
62 |
|
video3d/render/mlptexture.py
CHANGED
@@ -8,7 +8,7 @@
|
|
8 |
# its affiliates is strictly prohibited.
|
9 |
|
10 |
import torch
|
11 |
-
import tinycudann as tcnn
|
12 |
import numpy as np
|
13 |
|
14 |
#######################################################################################################################################################
|
@@ -118,5 +118,5 @@ class MLPTexture3D(torch.nn.Module):
|
|
118 |
|
119 |
return out.view(*texc.shape[:-1], self.channels) # Remap to [n, h, w, c]
|
120 |
|
121 |
-
def cleanup(self):
|
122 |
-
|
|
|
8 |
# its affiliates is strictly prohibited.
|
9 |
|
10 |
import torch
|
11 |
+
# import tinycudann as tcnn
|
12 |
import numpy as np
|
13 |
|
14 |
#######################################################################################################################################################
|
|
|
118 |
|
119 |
return out.view(*texc.shape[:-1], self.channels) # Remap to [n, h, w, c]
|
120 |
|
121 |
+
# def cleanup(self):
|
122 |
+
# tcnn.free_temporary_memory()
|