hugoycj commited on
Commit
22ec042
1 Parent(s): e685b72

fix: Update load_model

Browse files
Files changed (1) hide show
  1. dust3r/model.py +1 -1
dust3r/model.py CHANGED
@@ -80,7 +80,7 @@ class AsymmetricCroCo3DStereo (
80
  @classmethod
81
  def from_pretrained(cls, pretrained_model_name_or_path, **kw):
82
  if os.path.isfile(pretrained_model_name_or_path) or urllib.parse.urlparse(pretrained_model_name_or_path).scheme in ('http', 'https'):
83
- return load_model(pretrained_model_name_or_path, device='cpu', landscape_only=kw['landscape_only'])
84
  else:
85
  return super(AsymmetricCroCo3DStereo, cls).from_pretrained(pretrained_model_name_or_path, **kw)
86
 
 
80
  @classmethod
81
  def from_pretrained(cls, pretrained_model_name_or_path, **kw):
82
  if os.path.isfile(pretrained_model_name_or_path) or urllib.parse.urlparse(pretrained_model_name_or_path).scheme in ('http', 'https'):
83
+ return load_model(pretrained_model_name_or_path, device='cpu', **kw)
84
  else:
85
  return super(AsymmetricCroCo3DStereo, cls).from_pretrained(pretrained_model_name_or_path, **kw)
86