Updating configs for compatibility

config.json

@@ -1,16 +1,22 @@
 {
-  "_name_or_path": "neuralbioinfo/prokbert-mini",
   "architectures": [
     "ProkBertForMaskedLM"
   ],
   "attention_probs_dropout_prob": 0.1,
   "auto_map": {
-    "AutoConfig": "models.ProkBertConfig",
-    "AutoModel": "neuralbioinfo/prokbert-mini--models.ProkBertModel",
-    "AutoModelForMaskedLM": "models.ProkBertForMaskedLM",
-    "AutoModelForSequenceClassification": "neuralbioinfo/prokbert-mini--models.ProkBertForSequenceClassification"
+    "AutoConfig": "neuralbioinfo/nbrg-transformers--models.ProkBertConfig",
+    "AutoModel": "neuralbioinfo/nbrg-transformers--models.ProkBertModel",
+    "AutoModelForMaskedLM": "neuralbioinfo/nbrg-transformers--models.ProkBertForMaskedLM",
+    "AutoModelForSequenceClassification": "neuralbioinfo/nbrg-transformers--models.ProkBertForSequenceClassification"
   },
-  "classification_dropout_rate": 0.1,
+  "classifier_dropout": 0.1,
+  "classifier_head_type": "linear",
+  "classifier_mlp_hidden_size": null,
+  "classifier_pooling": "attention",
+  "curricular_embedding_size": null,
+  "curricular_margin": 0.5,
+  "curricular_scale": 64.0,
+  "dtype": "float32",
   "hidden_act": "gelu",
   "hidden_dropout_prob": 0.1,
   "hidden_size": 384,
@@ -25,8 +31,7 @@
   "pad_token_id": 0,
   "position_embedding_type": "relative_key_query",
   "shift": 1,
-  "torch_dtype": "float32",
-  "transformers_version": "4.48.0.dev0",
+  "transformers_version": "4.57.6",
   "type_vocab_size": 2,
   "use_cache": true,
   "vocab_size": 4101

generation_config.json

@@ -1,5 +1,5 @@
 {
   "_from_model_config": true,
   "pad_token_id": 0,
-  "transformers_version": "4.48.0.dev0"
+  "transformers_version": "4.57.6"
 }

tokenizer_config.json

@@ -41,12 +41,6 @@
       "special": true
     }
   },
-  "auto_map": {
-    "AutoTokenizer": [
-      "tokenizer.LCATokenizer",
-      null
-    ]
-  },
   "clean_up_tokenization_spaces": false,
   "cls_token": "[CLS]",
   "extra_special_tokens": {},
@@ -55,6 +49,9 @@
   "pad_token": "[PAD]",
   "sep_token": "[SEP]",
   "tokenizer_class": "LCATokenizer",
+  "auto_map": {
+    "AutoTokenizer": ["neuralbioinfo/nbrg-transformers--tokenizer.LCATokenizer", null]
+  },
   "unk_token": "[UNK]",
   "kmer": 6,
   "shift": 1,