faneggg commited on
Commit
f332442
·
verified ·
1 Parent(s): 9037d8e

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -37,7 +37,7 @@ from dust3r.demo import _convert_scene_output_to_glb
37
  def get_dust3r_args_parser():
38
  parser = argparse.ArgumentParser()
39
  parser.add_argument("--image_size", type=int, default=512, choices=[512, 224], help="image size")
40
- parser.add_argument("--model_path", type=str, default="endless-ai/dust3r/DUSt3R_ViTLarge_BaseDecoder_512_dpt.pth", help="path to the model weights")
41
  parser.add_argument("--device", type=str, default='cuda', help="pytorch device")
42
  parser.add_argument("--batch_size", type=int, default=1)
43
  parser.add_argument("--schedule", type=str, default='linear')
 
37
  def get_dust3r_args_parser():
38
  parser = argparse.ArgumentParser()
39
  parser.add_argument("--image_size", type=int, default=512, choices=[512, 224], help="image size")
40
+ parser.add_argument("--model_path", type=str, default="naver/DUSt3R_ViTLarge_BaseDecoder_512_dpt", help="path to the model weights")
41
  parser.add_argument("--device", type=str, default='cuda', help="pytorch device")
42
  parser.add_argument("--batch_size", type=int, default=1)
43
  parser.add_argument("--schedule", type=str, default='linear')