Add auto_map to config.json

config.json

@@ -22,6 +22,7 @@
   "transformers_version": "4.49.0",
   "auto_map": {
     "AutoConfig": "configuration_nicheformer.NicheformerConfig",
+    "AutoModelForMaskedLM": "modeling_nicheformer.NicheformerForHuggingFace",
     "AutoModel": "modeling_nicheformer.NicheformerForHuggingFace",
     "AutoTokenizer": "tokenization_nicheformer.NicheformerTokenizer"
   }