yansong1616 commited on
Commit
5747970
1 Parent(s): eeb12d4

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -69,7 +69,7 @@ def get_args_parser():
69
  "If None, will search for an available port starting at 7860."),
70
  default=None)
71
  parser.add_argument("--weights", type=str, default="./checkpoints/DUSt3R_ViTLarge_BaseDecoder_512_dpt.pth", required=False, help="path to the model weights")
72
- parser.add_argument("--device", type=str, default='cuda', help="pytorch device")
73
  parser.add_argument("--tmp_dir", type=str, default=None, help="value for tempfile.tempdir")
74
  return parser
75
 
 
69
  "If None, will search for an available port starting at 7860."),
70
  default=None)
71
  parser.add_argument("--weights", type=str, default="./checkpoints/DUSt3R_ViTLarge_BaseDecoder_512_dpt.pth", required=False, help="path to the model weights")
72
+ parser.add_argument("--device", type=str, default='cpu', help="pytorch device")
73
  parser.add_argument("--tmp_dir", type=str, default=None, help="value for tempfile.tempdir")
74
  return parser
75