Upload model

configuration_bionexttager.py

@@ -24,13 +24,5 @@ class BioNextTaggerConfig(PretrainedConfig):
         self.crf_reduction = crf_reduction
         super().__init__(**kwargs)
         
-    def get_backbonemodel_config(self):
-        print(f"model_path {self._name_or_path}", flush=True)
-        backbonemodel_cfg = AutoConfig.from_pretrained(self._name_or_path,
-                                                       trust_remote_code=True)#.to_dict()
-        for k in backbonemodel_cfg.to_dict():
-            if hasattr(self, k):
-                setattr(backbonemodel_cfg,k, getattr(self,k))
-        #print("BACKend config end?", flush=True)
-        return backbonemodel_cfg
+
         

modeling_bionexttagger.py

@@ -23,7 +23,7 @@ class BioNextTaggerModel(PreTrainedModel):
         super().__init__(config)
         self.num_labels = config.num_labels
         #print("LOAD BERT?", flush=True)
-        self.bert = BertModel(config.get_backbonemodel_config(), add_pooling_layer=False)
+        self.bert = BertModel(config, add_pooling_layer=False)
                     #AutoModel.from_pretrained(config._name_or_path, 
                     #                          config=config.get_backbonemodel_config(), 
                     #                          add_pooling_layer=False)