zpn commited on
Commit
084721d
1 Parent(s): 3916676

Update config.json

Browse files
Files changed (1) hide show
  1. config.json +4 -3
config.json CHANGED
@@ -4,9 +4,10 @@
4
  "NomicBertModel"
5
  ],
6
  "attn_pdrop": 0.0,
7
- "auto_map": {
8
- "AutoConfig": "configuration_hf_nomic_bert.NomicBertConfig",
9
- "AutoModel": "modeling_hf_nomic_bert.NomicBertModel"
 
10
  },
11
  "bos_token_id": null,
12
  "causal": false,
 
4
  "NomicBertModel"
5
  ],
6
  "attn_pdrop": 0.0,
7
+ "auto_map": {
8
+ "AutoConfig": "nomic-ai/nomic-bert-2048--configuration_hf_nomic_bert.NomicBertConfig",
9
+ "AutoModel": "nomic-ai/nomic-bert-2048--modeling_hf_nomic_bert.NomicBertModel",
10
+ "AutoModelForMaskedLM": "nomic-ai/nomic-bert-2048--modeling_hf_nomic_bert.NomicBertForPreTraining"
11
  },
12
  "bos_token_id": null,
13
  "causal": false,