diff --git a/config.json b/config.json
index b51379b..e43ecf4 100644
--- a/config.json
+++ b/config.json
@@ -3,6 +3,10 @@
   "architectures": [
     "PhiForCausalLM"
   ],
+  "auto_map": {
+    "AutoConfig": "configuration_phi.PhiConfig",
+    "AutoModelForCausalLM": "modeling_phi.PhiForCausalLM"
+  },
   "attention_dropout": 0.0,
   "bos_token_id": null,
   "embd_pdrop": 0.0,