Spaces:

xinjjj
/

ImgRoboAssetGen

Running on Zero

xinjie.wang commited on 24 days ago

Commit

4cd8e70

1 Parent(s): 8dc4043

update

Files changed (1) hide show

common.py CHANGED Viewed

@@ -498,7 +498,7 @@ def extract_3d_representations_v2(
     print("debug1")
     output_root = TMP_DIR
     user_dir = os.path.join(output_root, str(req.session_hash))
-    gs_model, mesh_model = unpack_state(state, device="cpu")
     print("debug2")
     filename = "sample"

     print("debug1")
     output_root = TMP_DIR
     user_dir = os.path.join(output_root, str(req.session_hash))
+    gs_model, mesh_model = unpack_state(state, device="cuda")
     print("debug2")
     filename = "sample"