xinjie.wang commited on
Commit
c9bd2e0
·
1 Parent(s): ce1512c
Files changed (1) hide show
  1. common.py +1 -1
common.py CHANGED
@@ -501,7 +501,7 @@ def extract_3d_representations_v2(
501
  print("debug1")
502
  output_root = TMP_DIR
503
  user_dir = os.path.join(output_root, str(req.session_hash))
504
- gs_model, mesh_model = unpack_state(state, device="cuda")
505
  print("debug2")
506
 
507
  filename = "sample"
 
501
  print("debug1")
502
  output_root = TMP_DIR
503
  user_dir = os.path.join(output_root, str(req.session_hash))
504
+ gs_model, mesh_model = unpack_state(state, device="cpu")
505
  print("debug2")
506
 
507
  filename = "sample"