Update config.json
Browse files- config.json +4 -4
config.json
CHANGED
@@ -4,9 +4,9 @@
|
|
4 |
"BiModernVBert"
|
5 |
],
|
6 |
"auto_map": {
|
7 |
-
"AutoConfig": "
|
8 |
-
"AutoModel": "
|
9 |
-
"AutoModelForMaskedLM": "
|
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": "
|
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,
|