Upload BERTClassifier

config.json

@@ -22,7 +22,7 @@
   "pad_token_id": 0,
   "position_embedding_type": "absolute",
   "torch_dtype": "float32",
-  "transformers_version": "4.33.3",
+  "transformers_version": "4.46.1",
   "type_vocab_size": 2,
   "use_cache": true,
   "vocab_size": 30522

model.py

@@ -1,5 +1,18 @@
 from torch import nn
-from transformers import BertModel, PreTrainedModel, BertConfig
+from easydict import EasyDict as MyEasyDict
+from transformers import BertModel, PreTrainedModel, BertConfig, PretrainedConfig
+
+
+class BertConfig(PretrainedConfig):
+    model_type = "bert"
+
+    def __init__(
+            self,
+            model_config=None,
+            **kwargs):
+        super().__init__(**kwargs)
+        self.model_config  = MyEasyDict(model_config)
+
 
 class BERTClassifier(PreTrainedModel):
     config_class = BertConfig

model.safetensors

@@ -1,3 +1,3 @@
 version https://git-lfs.github.com/spec/v1
-oid sha256:cd87541ad4896795c0e56e29cd5481f8014fbbf7f530de7af98edc59196d5100
+oid sha256:5f99e4511b96ce471345446693e579edb7636a316c72efc76dedb34d579f4de7
 size 438001696