magicslabnu
/

OutEffHop_bert_base

Model card Files Files and versions

robinzixuan commited on Jun 7, 2024

Commit

c825946

·

verified ·

1 Parent(s): 60ff2f2

Update config.json

Files changed (1) hide show

config.json +5 -4

config.json CHANGED Viewed

@@ -22,14 +22,15 @@
   "torch_dtype": "float32",
   "transformers_version": "4.31.0",
-  "type_vocab_size": 2,
   "use_cache": true,
   "vocab_size": 30522,
   "attn_implementation":"OutEffHop",
   "auto_map":
       {"AutoConfig": "configuration_bert.BertConfig",
-        "AutoModel": "bert_layers.BertModel",
-        "AutoModelForMaskedLM": "bert_layers.BertForMaskedLM",
-        "AutoModelForSequenceClassification": "bert_layers.BertForSequenceClassification"
       },
 }

   "torch_dtype": "float32",
   "transformers_version": "4.31.0",
+  "type_vocab_size": 2
+,
   "use_cache": true,
   "vocab_size": 30522,
   "attn_implementation":"OutEffHop",
   "auto_map":
       {"AutoConfig": "configuration_bert.BertConfig",
+        "AutoModel": "modeling_bert.BertModel",
+        "AutoModelForMaskedLM": "modeling_bert.BertForMaskedLM",
+        "AutoModelForSequenceClassification": "modeling_bert.BertForSequenceClassification"
       },
 }