diff --git a/dust3r/model.py b/dust3r/model.py index a245d4e..0eb8884 100644 --- a/dust3r/model.py +++ b/dust3r/model.py @@ -27,7 +27,7 @@ def load_model(model_path, device, verbose=True): if verbose: print('... loading model from', model_path) - ckpt = torch.load(model_path, map_location='cpu') + ckpt = torch.load(model_path, map_location='cpu', weights_only=False) args = ckpt['args'].model.replace("ManyAR_PatchEmbed", "PatchEmbedDust3R") if 'landscape_only' not in args: args = args[:-1] + ', landscape_only=False)'