From 2dec684b766fa7970914138ab4eead3ea6c60ef1 Mon Sep 17 00:00:00 2001
From: Yagil Burowski <yagil@elementlabs.ai>
Date: Fri, 22 Dec 2023 14:43:50 -0500
Subject: [PATCH] update schema

---
 schema.json | 2 +-
 1 file changed, 1 insertion(+), 1 deletion(-)

diff --git a/schema.json b/schema.json
index 89926a2..9b6cb0d 100644
--- a/schema.json
+++ b/schema.json
@@ -59,7 +59,7 @@
       },
       "arch": {
         "type": "string",
-        "enum": ["llama", "pythia", "gpt-neo-x", "gpt-j", "mpt", "replit", "starcoder", "falcon", "mistral", "stablelm"]
+        "enum": ["llama", "pythia", "gpt-neo-x", "gpt-j", "mpt", "replit", "starcoder", "falcon", "mistral", "stablelm", "phi-2"]
       },
       "description": {
         "type": "string"