xinjie.wang commited on
Commit
7c935af
·
1 Parent(s): 906626c
Files changed (1) hide show
  1. common.py +3 -2
common.py CHANGED
@@ -377,8 +377,9 @@ def image_to_3d(
377
 
378
  gs_model = outputs["gaussian"][0]
379
  mesh_model = outputs["mesh"][0]
380
- color_images = render_video(gs_model)["color"]
381
- normal_images = render_video(mesh_model)["normal"]
 
382
  output_root = TMP_DIR
383
  if req is not None:
384
  output_root = os.path.join(output_root, str(req.session_hash))
 
377
 
378
  gs_model = outputs["gaussian"][0]
379
  mesh_model = outputs["mesh"][0]
380
+ with torch.no_grad():
381
+ color_images = render_video(gs_model, num_frames=1)["color"]
382
+ normal_images = render_video(mesh_model, num_frames=1)["normal"]
383
  output_root = TMP_DIR
384
  if req is not None:
385
  output_root = os.path.join(output_root, str(req.session_hash))