Spaces:
Running
on
L40S
Running
on
L40S
update
Browse files- slrm/models/lrm_mesh.py +2 -2
slrm/models/lrm_mesh.py
CHANGED
@@ -116,13 +116,13 @@ class MeshSLRM(nn.Module):
|
|
116 |
camera = OrthogonalCamera(device=device)
|
117 |
|
118 |
with torch.cuda.amp.autocast(enabled=False):
|
119 |
-
renderer = NeuralRender(device, camera_model=camera)
|
120 |
self.geometry = FlexiCubesGeometry(
|
121 |
grid_res_xy=self.grid_res_xy,
|
122 |
grid_res_z=self.grid_res_z,
|
123 |
scale_xy=self.grid_scale_xy,
|
124 |
scale_z=self.grid_scale_z,
|
125 |
-
renderer=
|
126 |
render_type='neural_render',
|
127 |
device=device,
|
128 |
)
|
|
|
116 |
camera = OrthogonalCamera(device=device)
|
117 |
|
118 |
with torch.cuda.amp.autocast(enabled=False):
|
119 |
+
# renderer = NeuralRender(device, camera_model=camera)
|
120 |
self.geometry = FlexiCubesGeometry(
|
121 |
grid_res_xy=self.grid_res_xy,
|
122 |
grid_res_z=self.grid_res_z,
|
123 |
scale_xy=self.grid_scale_xy,
|
124 |
scale_z=self.grid_scale_z,
|
125 |
+
renderer=None,
|
126 |
render_type='neural_render',
|
127 |
device=device,
|
128 |
)
|