From 84a3f3a99beff218b60b5e633c282bfc9c704ed2 Mon Sep 17 00:00:00 2001
From: Yagil Burowski <yagil@elementlabs.ai>
Date: Wed, 21 Feb 2024 19:39:15 -0500
Subject: [PATCH] update schema to support gemma

---
 schema.json | 4 ++--
 1 file changed, 2 insertions(+), 2 deletions(-)

diff --git a/schema.json b/schema.json
index 12cf4e7..fcfc55f 100644
--- a/schema.json
+++ b/schema.json
@@ -51,7 +51,7 @@
       },
       "numParameters": {
         "type": "string",
-        "enum": ["1.5B", "3B", "4B", "6.7B", "7B", "13B", "15B", "30B", "65B", "unknown"]
+        "enum": ["1.5B", "2B", "3B", "4B", "6.7B", "7B", "13B", "15B", "30B", "65B", "unknown"]
       },
       "trainedFor": {
         "type": "string",
@@ -59,7 +59,7 @@
       },
       "arch": {
         "type": "string",
-        "enum": ["llama", "pythia", "gpt-neo-x", "gpt-j", "mpt", "replit", "starcoder", "falcon", "mistral", "stablelm", "phi2", "qwen2"]
+        "enum": ["llama", "pythia", "gpt-neo-x", "gpt-j", "mpt", "replit", "starcoder", "falcon", "mistral", "stablelm", "phi2", "qwen2", "gemma"]
       },
       "description": {
         "type": "string"