Spaces:
Paused
Paused
yansong1616
commited on
Commit
•
5747970
1
Parent(s):
eeb12d4
Update app.py
Browse files
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='
|
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 |
|