Spaces:

xinjjj
/

ImgRoboAssetGen

Running on Zero

xinjie.wang commited on Apr 21

Commit

c9bd2e0

1 Parent(s): ce1512c

update

Files changed (1) hide show

common.py CHANGED Viewed

@@ -501,7 +501,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="cuda")
     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="cpu")
     print("debug2")
     filename = "sample"