Upload config

config.json

@@ -1,11 +1,8 @@
 {
-  "architectures": [
-    "BertForSequenceClassification"
-  ],
+  "affine": true,
   "attention_probs_dropout_prob": 0.1,
   "auto_map": {
-    "AutoConfig": "configuration_bert.BertConfig",
-    "AutoModelForSequenceClassification": "modeling_bert.BertForSequenceClassification"
+    "AutoConfig": "configuration_bert.BertConfig"
   },
   "classifier_dropout": null,
   "hidden_act": "gelu",
@@ -20,7 +17,6 @@
   "num_hidden_layers": 12,
   "pad_token_id": 0,
   "position_embedding_type": "absolute",
-  "torch_dtype": "float32",
   "transformers_version": "4.33.3",
   "type_vocab_size": 2,
   "use_cache": true,

configuration_bert.py

@@ -23,6 +23,7 @@ class BertConfig(PretrainedConfig):
         position_embedding_type="absolute",
         use_cache=True,
         classifier_dropout=None,
+        affine=True, 
         **kwargs,
     ):
         super().__init__(pad_token_id=pad_token_id, **kwargs)
@@ -41,4 +42,5 @@ class BertConfig(PretrainedConfig):
         self.layer_norm_eps = layer_norm_eps
         self.position_embedding_type = position_embedding_type
         self.use_cache = use_cache
-        self.classifier_dropout = classifier_dropout
+        self.classifier_dropout = classifier_dropout
+        self.affine = affine