carlesoctav
/

sealion-bert

Inference Endpoints

Model card Files Files and versions Community

carlesoctav commited on Jun 19, 2024

Commit

8546a7e

·

verified ·

1 Parent(s): a4c6a10

Update config.json

Files changed (1) hide show

config.json +2 -1

config.json CHANGED Viewed

@@ -7,7 +7,8 @@
   "attention_probs_dropout_prob": 0.0,
   "auto_map": {
     "AutoConfig": "aisingapore/sealion-bert-base--configuration_bert.BertConfig",
-    "AutoModelForMaskedLM": "aisingapore/sealion-bert-base--bert_layers.BertForMaskedLM"
   },
   "classifier_dropout": null,
   "gradient_checkpointing": false,

   "attention_probs_dropout_prob": 0.0,
   "auto_map": {
     "AutoConfig": "aisingapore/sealion-bert-base--configuration_bert.BertConfig",
+    "AutoModelForMaskedLM": "aisingapore/sealion-bert-base--bert_layers.BertForMaskedLM",
+    "AutoModel": "aisingapore/sealion-bert-base--bert_layers.BertModel"
   },
   "classifier_dropout": null,
   "gradient_checkpointing": false,