Spaces:
Running
on
Zero
Running
on
Zero
Stanislaw Szymanowicz
commited on
Commit
•
37d5e7e
1
Parent(s):
9f405ed
Map model to cpu
Browse files
app.py
CHANGED
|
@@ -50,7 +50,7 @@ def main():
|
|
| 50 |
|
| 51 |
model = GaussianSplatPredictor(model_cfg)
|
| 52 |
|
| 53 |
-
ckpt_loaded = torch.load(model_path, map_location=
|
| 54 |
model.load_state_dict(ckpt_loaded["model_state_dict"])
|
| 55 |
model.to(device)
|
| 56 |
|
|
|
|
| 50 |
|
| 51 |
model = GaussianSplatPredictor(model_cfg)
|
| 52 |
|
| 53 |
+
ckpt_loaded = torch.load(model_path, map_location="cpu")
|
| 54 |
model.load_state_dict(ckpt_loaded["model_state_dict"])
|
| 55 |
model.to(device)
|
| 56 |
|