Spaces:
Sleeping
Sleeping
hugoycj
commited on
Commit
•
22ec042
1
Parent(s):
e685b72
fix: Update load_model
Browse files- 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',
|
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 |
|