Vincentqyw
commited on
Commit
·
d7ac6a3
1
Parent(s):
0c624e7
update: basemodel
Browse files- hloc/matchers/aspanformer.py +0 -1
- hloc/utils/base_model.py +0 -1
hloc/matchers/aspanformer.py
CHANGED
@@ -43,7 +43,6 @@ class ASpanFormer(BaseModel):
|
|
43 |
Path(__file__).stem, self.conf["model_name"]
|
44 |
),
|
45 |
)
|
46 |
-
|
47 |
state_dict = torch.load(str(model_path), map_location="cpu")[
|
48 |
"state_dict"
|
49 |
]
|
|
|
43 |
Path(__file__).stem, self.conf["model_name"]
|
44 |
),
|
45 |
)
|
|
|
46 |
state_dict = torch.load(str(model_path), map_location="cpu")[
|
47 |
"state_dict"
|
48 |
]
|
hloc/utils/base_model.py
CHANGED
@@ -34,7 +34,6 @@ class BaseModel(nn.Module, metaclass=ABCMeta):
|
|
34 |
"""To be implemented by the child class."""
|
35 |
raise NotImplementedError
|
36 |
|
37 |
-
@abstractmethod
|
38 |
def _download_model(self, repo_id=None, filename=None, **kwargs):
|
39 |
"""Download model from hf hub and return the path."""
|
40 |
return hf_hub_download(
|
|
|
34 |
"""To be implemented by the child class."""
|
35 |
raise NotImplementedError
|
36 |
|
|
|
37 |
def _download_model(self, repo_id=None, filename=None, **kwargs):
|
38 |
"""Download model from hf hub and return the path."""
|
39 |
return hf_hub_download(
|