Dont move viewer outputs to cpu (#1335)

This commit is contained in:
Matthew Tancik authored and GitHub committed 2023-02-01 17:33:06 -08:00
1 parent db83068fcd
commit 3a416a387b
1 file changed
-2
-2
View File
@@ -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