T-Almeida commited on
Commit
4fd0c83
·
verified ·
1 Parent(s): 3064e9c

Upload model

Browse files
Files changed (1) hide show
  1. configuration_bionexttager.py +4 -2
configuration_bionexttager.py CHANGED
@@ -25,10 +25,12 @@ class BioNextTaggerConfig(PretrainedConfig):
25
  super().__init__(**kwargs)
26
 
27
  def get_backbonemodel_config(self):
28
- backbonemodel_cfg = AutoConfig.from_pretrained(self._name_or_path)#.to_dict()
 
 
29
  for k in backbonemodel_cfg.to_dict():
30
  if hasattr(self, k):
31
  setattr(backbonemodel_cfg,k, getattr(self,k))
32
-
33
  return backbonemodel_cfg
34
 
 
25
  super().__init__(**kwargs)
26
 
27
  def get_backbonemodel_config(self):
28
+ print("BACKend config start?", flush=True)
29
+ backbonemodel_cfg = AutoConfig.from_pretrained(self._name_or_path,
30
+ trust_remote_code=True)#.to_dict()
31
  for k in backbonemodel_cfg.to_dict():
32
  if hasattr(self, k):
33
  setattr(backbonemodel_cfg,k, getattr(self,k))
34
+ print("BACKend config end?", flush=True)
35
  return backbonemodel_cfg
36