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"