Spaces:
Running
on
Zero
Running
on
Zero
xinjie.wang
commited on
Commit
·
ce1512c
1
Parent(s):
b760e81
update
Browse files
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)
|