Hugging Face's logo

Spaces:
szymanowiczs
/
splatter_image
Runtime error

App Files Community
1
Stanislaw Szymanowicz commited on
Commit
37d5e7e
·
1 Parent(s): 9f405ed

Map model to cpu

Browse files
Files changed (1) hide show
  1. app.py +1 -1
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=device)
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