remg1997 commited on
Commit
516fd02
·
verified ·
1 Parent(s): 4784dd3

Add missing AutoModel auto_map entry (needed for babylm-eval finetune.run's AutoModel.from_pretrained)

Browse files
Files changed (1) hide show
  1. config.json +3 -2
config.json CHANGED
@@ -6,7 +6,8 @@
6
  "attention_pattern": "alternating",
7
  "auto_map": {
8
  "AutoConfig": "configuration_modernbert_small.ModernBertSmallConfig",
9
- "AutoModelForMaskedLM": "modeling_modernbert_small.ModernBertSmallForMaskedLM"
 
10
  },
11
  "compositional_bottleneck_dim": 128,
12
  "compositional_frequency_shuffle_seed": 42,
@@ -61098,4 +61099,4 @@
61098
  ],
61099
  "transformers_version": "4.57.6",
61100
  "vocab_size": 30522
61101
- }
 
6
  "attention_pattern": "alternating",
7
  "auto_map": {
8
  "AutoConfig": "configuration_modernbert_small.ModernBertSmallConfig",
9
+ "AutoModelForMaskedLM": "modeling_modernbert_small.ModernBertSmallForMaskedLM",
10
+ "AutoModel": "modeling_modernbert_small.ModernBertSmallModel"
11
  },
12
  "compositional_bottleneck_dim": 128,
13
  "compositional_frequency_shuffle_seed": 42,
 
61099
  ],
61100
  "transformers_version": "4.57.6",
61101
  "vocab_size": 30522
61102
+ }