xinjie.wang commited on
Commit
ce1512c
·
1 Parent(s): b760e81
Files changed (1) hide show
  1. common.py +1 -0
common.py CHANGED
@@ -332,6 +332,7 @@ def unpack_state(state: dict, device: str = "cuda") -> tuple[Gaussian, dict]:
332
  scaling_bias=state["gaussian"]["scaling_bias"],
333
  opacity_bias=state["gaussian"]["opacity_bias"],
334
  scaling_activation=state["gaussian"]["scaling_activation"],
 
335
  )
336
  print("debug12")
337
  gs._xyz = torch.tensor(state["gaussian"]["_xyz"], device=device)
 
332
  scaling_bias=state["gaussian"]["scaling_bias"],
333
  opacity_bias=state["gaussian"]["opacity_bias"],
334
  scaling_activation=state["gaussian"]["scaling_activation"],
335
+ device="cpu",
336
  )
337
  print("debug12")
338
  gs._xyz = torch.tensor(state["gaussian"]["_xyz"], device=device)