From 051d15f1e771cfdb09228a4a4ba72e6d22e9b136 Mon Sep 17 00:00:00 2001
From: Gustavo de Rosa <gugarosa@users.noreply.huggingface.co>
Date: Thu, 11 Jan 2024 11:22:42 +0000
Subject: [PATCH] Update config.json

---
 config.json | 4 ++++
 1 file changed, 4 insertions(+)

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,