Upload model

config.json

@@ -1,14 +1,14 @@
 {
   "_name_or_path": "michiyasunaga/BioLinkBERT-large",
   "architectures": [
-    "BioNextTaggerModel"
+    "BioNExtTaggerModel"
   ],
   "args_random_seed": 42,
   "attention_probs_dropout_prob": 0.1,
   "augmentation": "unk",
   "auto_map": {
-    "AutoConfig": "configuration_bionexttager.BioNextTaggerConfig",
-    "AutoModel": "modeling_bionexttagger.BioNextTaggerModel"
+    "AutoConfig": "configuration_bionexttager.BioNExtTaggerConfig",
+    "AutoModel": "modeling_bionexttagger.BioNExtTaggerModel"
   },
   "classifier_dropout": null,
   "context_size": 2,

configuration_bionexttager.py

@@ -3,7 +3,7 @@ from transformers import PretrainedConfig, AutoConfig
 from typing import List
 
 
-class BioNextTaggerConfig(PretrainedConfig):
+class BioNExtTaggerConfig(PretrainedConfig):
     model_type = "crf-tagger"
 
     def __init__(

modeling_bionexttagger.py

@@ -10,13 +10,13 @@ from typing import List, Optional
 
 import torch
 from itertools import islice
-from .configuration_bionexttager import BioNextTaggerConfig
+from .configuration_bionexttager import BioNExtTaggerConfig
 
 
 NUM_PER_LAYER = 16
 
-class BioNextTaggerModel(PreTrainedModel):
-    config_class = BioNextTaggerConfig
+class BioNExtTaggerModel(PreTrainedModel):
+    config_class = BioNExtTaggerConfig
     _keys_to_ignore_on_load_unexpected = [r"pooler"]
 
     def __init__(self, config):