From 3a416a387bf0a145874826ef32c05ea795c0b290 Mon Sep 17 00:00:00 2001 From: Matthew Tancik Date: Wed, 1 Feb 2023 17:33:06 -0800 Subject: [PATCH] Dont move viewer outputs to cpu (#1335) --- nerfstudio/viewer/server/viewer_utils.py | 2 -- 1 file changed, 2 deletions(-) diff --git a/nerfstudio/viewer/server/viewer_utils.py b/nerfstudio/viewer/server/viewer_utils.py index 44056537f..1a3fefd51 100644 --- a/nerfstudio/viewer/server/viewer_utils.py +++ b/nerfstudio/viewer/server/viewer_utils.py @@ -48,7 +48,6 @@ from nerfstudio.models.base_model import Model from nerfstudio.utils import colormaps, profiler, writer from nerfstudio.utils.decorators import check_main_thread, decorate_all from nerfstudio.utils.io import load_from_json, write_to_json -from nerfstudio.utils.misc import get_dict_to_torch from nerfstudio.utils.writer import GLOBAL_BUFFER, EventName, TimeWriter from nerfstudio.viewer.server.subprocess import run_viewer_bridge_server_as_subprocess from nerfstudio.viewer.server.utils import ( @@ -163,7 +162,6 @@ class RenderThread(threading.Thread): self.exc = e if outputs: - outputs = get_dict_to_torch(outputs) self.vis_outputs = outputs self.state.check_done_render = True