Add auto_map to config.json for Auto classes compatibility

config.json

@@ -22,8 +22,8 @@
   "transformers_version": "4.49.0",
   "auto_map": {
     "AutoConfig": "configuration_nicheformer.NicheformerConfig",
-    "AutoModelForMaskedLM": "modeling_nicheformer.NicheformerForHuggingFace",
-    "AutoModel": "modeling_nicheformer.NicheformerForHuggingFace",
+    "AutoModelForMaskedLM": "modeling_nicheformer.NicheformerForMaskedLM",
+    "AutoModel": "modeling_nicheformer.NicheformerModel",
     "AutoTokenizer": [
       "tokenization_nicheformer.NicheformerTokenizer",
       null