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

config.json

@@ -6,7 +6,8 @@
   "attention_pattern": "alternating",
   "auto_map": {
     "AutoConfig": "configuration_modernbert_small.ModernBertSmallConfig",
-    "AutoModelForMaskedLM": "modeling_modernbert_small.ModernBertSmallForMaskedLM"
+    "AutoModelForMaskedLM": "modeling_modernbert_small.ModernBertSmallForMaskedLM",
+    "AutoModel": "modeling_modernbert_small.ModernBertSmallModel"
   },
   "dtype": "float32",
   "embedding_dropout": 0.0,
@@ -30,4 +31,4 @@
   "tie_word_embeddings": false,
   "transformers_version": "4.57.6",
   "vocab_size": 30522
-}
+}