admin commited on
Commit
dfca8bb
1 Parent(s): c893964

fix trust_remote_code=True,

Browse files
Files changed (1) hide show
  1. model.py +5 -1
model.py CHANGED
@@ -38,7 +38,11 @@ class EvalNet:
38
  return backbone_list[0]
39
 
40
  def _model_info(self, m_ver: str):
41
- backbone_list = MsDataset.load("monetjoe/cv_backbones", split="v1")
 
 
 
 
42
  backbone = self._get_backbone(m_ver, backbone_list)
43
  m_type = str(backbone["type"])
44
  input_size = int(backbone["input_size"])
 
38
  return backbone_list[0]
39
 
40
  def _model_info(self, m_ver: str):
41
+ backbone_list = MsDataset.load(
42
+ "monetjoe/cv_backbones",
43
+ split="v1",
44
+ trust_remote_code=True,
45
+ )
46
  backbone = self._get_backbone(m_ver, backbone_list)
47
  m_type = str(backbone["type"])
48
  input_size = int(backbone["input_size"])