YulianSa commited on
Commit
5e1cec3
·
1 Parent(s): f32c5b1
Files changed (1) hide show
  1. 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=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
  )