fix: Fix mast3r

mast3r/model.py

@@ -13,15 +13,17 @@ from mast3r.catmlp_dpt_head import mast3r_head_factory
 import mast3r.utils.path_to_dust3r  # noqa
 from dust3r.model import AsymmetricCroCo3DStereo  # noqa
 from dust3r.utils.misc import transpose_to_landscape  # noqa
-
+import urllib
 
 inf = float('inf')
 
 
-def load_model(model_path, device, verbose=True):
+def load_model(model_url, device, landscape_only=False, verbose=True):
     if verbose:
-        print('... loading model from', model_path)
-    ckpt = torch.load(model_path, map_location='cpu')
+        print('... loading model from', model_url)
+    
+    ckpt = torch.hub.load_state_dict_from_url(model_url, map_location='cpu', progress=verbose)
+    
     args = ckpt['args'].model.replace("ManyAR_PatchEmbed", "PatchEmbedDust3R")
     if 'landscape_only' not in args:
         args = args[:-1] + ', landscape_only=False)'
@@ -46,10 +48,10 @@ class AsymmetricMASt3R(AsymmetricCroCo3DStereo):
 
     @classmethod
     def from_pretrained(cls, pretrained_model_name_or_path, **kw):
-        if os.path.isfile(pretrained_model_name_or_path):
-            return load_model(pretrained_model_name_or_path, device='cpu')
+        if os.path.isfile(pretrained_model_name_or_path) or urllib.parse.urlparse(pretrained_model_name_or_path).scheme in ('http', 'https'):
+            return load_model(pretrained_model_name_or_path, device='cpu',  **kw)
         else:
-            return super(AsymmetricMASt3R, cls).from_pretrained(pretrained_model_name_or_path, **kw)
+            return super(AsymmetricCroCo3DStereo, cls).from_pretrained(pretrained_model_name_or_path, **kw)
 
     def set_downstream_head(self, output_mode, head_type, landscape_only, depth_mode, conf_mode, patch_size, img_size, **kw):
         assert img_size[0] % patch_size == 0 and img_size[