Ebtihal commited on
Commit
1ec7551
1 Parent(s): f4803db
Files changed (1) hide show
  1. config.json +2 -1
config.json CHANGED
@@ -1,7 +1,8 @@
1
  {
 
2
  "add_cross_attention": true,
3
  "architectures": [
4
- "BertLMHeadModel"
5
  ],
6
  "attention_probs_dropout_prob": 0.1,
7
  "gradient_checkpointing": false,
1
  {
2
+ "_name_or_path": "Ebtihal/Aurora",
3
  "add_cross_attention": true,
4
  "architectures": [
5
+ "BertForMaskedLM"
6
  ],
7
  "attention_probs_dropout_prob": 0.1,
8
  "gradient_checkpointing": false,