minor
Browse files
app.py
CHANGED
|
@@ -31,7 +31,7 @@ torch.backends.cuda.matmul.allow_tf32 = True
|
|
| 31 |
batch_size = 1
|
| 32 |
|
| 33 |
weights_path = "naver/" + 'MASt3R_ViTLarge_BaseDecoder_512_catmlpdpt_metric'
|
| 34 |
-
device = 'cuda'
|
| 35 |
model = AsymmetricMASt3R.from_pretrained(weights_path).to(device)
|
| 36 |
chkpt_tag = hash_md5(weights_path)
|
| 37 |
|
|
|
|
| 31 |
batch_size = 1
|
| 32 |
|
| 33 |
weights_path = "naver/" + 'MASt3R_ViTLarge_BaseDecoder_512_catmlpdpt_metric'
|
| 34 |
+
device = 'cuda' if torch.cuda.is_available() else 'cpu'
|
| 35 |
model = AsymmetricMASt3R.from_pretrained(weights_path).to(device)
|
| 36 |
chkpt_tag = hash_md5(weights_path)
|
| 37 |
|