Vincentqyw commited on
Commit
d7ac6a3
·
1 Parent(s): 0c624e7

update: basemodel

Browse files
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(