brandonsmart commited on
Commit
e059ab1
1 Parent(s): 7dfcb4e

Attempt to resolve CUDA issue in spaces

Browse files
Files changed (1) hide show
  1. demo.py +2 -1
demo.py CHANGED
@@ -55,7 +55,8 @@ if __name__ == '__main__':
55
  model_name = "brandonsmart/splatt3r_v1.0"
56
  filename = "epoch=19-step=1200.ckpt"
57
  weights_path = hf_hub_download(repo_id=model_name, filename=filename)
58
- model = main.MAST3RGaussians.load_from_checkpoint(weights_path, device)
 
59
  chkpt_tag = hash_md5(weights_path)
60
 
61
  # Define example inputs and their corresponding precalculated outputs
 
55
  model_name = "brandonsmart/splatt3r_v1.0"
56
  filename = "epoch=19-step=1200.ckpt"
57
  weights_path = hf_hub_download(repo_id=model_name, filename=filename)
58
+ model = main.MAST3RGaussians.load_from_checkpoint(weights_path, 'cpu')
59
+ model = model.to(device)
60
  chkpt_tag = hash_md5(weights_path)
61
 
62
  # Define example inputs and their corresponding precalculated outputs