Update nerfstudio/viewer/server/render_state_machine.py

Co-authored-by: Paul Wais <paulwais@gmail.com>
This commit is contained in:
Justin KerrandPaul Wais authored and GitHub committed 2023-10-16 08:59:32 -07:00
1 parent d46aede7d5
commit 46fde7c9e6
1 file changed
+1 -1
@@ -157,7 +157,7 @@ class RenderStateMachine(threading.Thread):
else:
with torch.no_grad():
if isGaussianSplattingModel:
outputs = self.viewer.get_model().get_outputs_for_camera_ray_bundle(camera)
outputs = self.viewer.get_model().get_outputs_for_camera_ray_bundle(camera_ray_bundle, camera=camera)
else:
outputs = self.viewer.get_model().get_outputs_for_camera_ray_bundle(camera_ray_bundle)
self.viewer.get_model().train()