update

common.py

@@ -323,7 +323,8 @@ def pack_state(gs: Gaussian, mesh: MeshExtractResult) -> dict:
     }
 
 
-def unpack_state(state: dict, device: str = "cuda") -> tuple[Gaussian, dict]:
+@spaces.GPU
+def unpack_state(state: dict, device: str = "cpu") -> tuple[Gaussian, dict]:
     print("debug11")
     gs = Gaussian(
         aabb=state["gaussian"]["aabb"],
@@ -332,7 +333,6 @@ def unpack_state(state: dict, device: str = "cuda") -> tuple[Gaussian, dict]:
         scaling_bias=state["gaussian"]["scaling_bias"],
         opacity_bias=state["gaussian"]["opacity_bias"],
         scaling_activation=state["gaussian"]["scaling_activation"],
-        device="cpu",
     )
     print("debug12")
     gs._xyz = torch.tensor(state["gaussian"]["_xyz"], device=device)