update

common.py

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