paultltc commited on
Commit
a24fee1
·
verified ·
1 Parent(s): e10d60d

Update config.json

Browse files
Files changed (1) hide show
  1. config.json +4 -4
config.json CHANGED
@@ -4,9 +4,9 @@
4
  "BiModernVBert"
5
  ],
6
  "auto_map": {
7
- "AutoConfig": "configuration_vbert.VBertConfig",
8
- "AutoModel": "modeling_vbert.VBertModel",
9
- "AutoModelForMaskedLM": "modeling_vbert.VBertForMaskedLM"
10
  },
11
  "freeze_config": {
12
  "freeze_lm_head": true,
@@ -17,7 +17,7 @@
17
  "image_token_id": 50407,
18
  "initializer_range": 0.02,
19
  "max_position_embeddings": 8192,
20
- "model_type": "vbert",
21
  "neftune_noise_alpha": 0.0,
22
  "output_attentions": false,
23
  "pixel_shuffle_factor": 4,
 
4
  "BiModernVBert"
5
  ],
6
  "auto_map": {
7
+ "AutoConfig": "configuration_modernvbert.ModernVBertConfig",
8
+ "AutoModel": "modeling_modernvbert.ModernVBertModel",
9
+ "AutoModelForMaskedLM": "modeling_modernvbert.ModernVBertForMaskedLM"
10
  },
11
  "freeze_config": {
12
  "freeze_lm_head": true,
 
17
  "image_token_id": 50407,
18
  "initializer_range": 0.02,
19
  "max_position_embeddings": 8192,
20
+ "model_type": "modernvbert",
21
  "neftune_noise_alpha": 0.0,
22
  "output_attentions": false,
23
  "pixel_shuffle_factor": 4,