config: fix it

config.json

@@ -3,7 +3,7 @@
   "architectures": [
     "xLSTMForCausalLM"
   ],
-  "transformers_version": "4.44.0",
+  "transformers_version": "4.44.1",
   "_xlstm_config": {
     "num_blocks": 24,
     "embedding_dim": 768,
@@ -20,5 +20,10 @@
   "vocab_size": 32000,
   "embedding_dim": 768,
   "context_length": 512,
-  "model_type": "xlstm"
-}
+  "model_type": "xlstm",
+  "auto_map": {
+    "AutoConfig": "configuration_xlstm.xLSTMConfig",
+    "AutoModelForCausalLM": "modeling_xlstm.xLSTMForCausalLM",
+    "AutoModel": "modeling_xlstm.xLSTMModel"
+  }
+}