update schema to support gemma

lmstudio-ai · Feb 22, 2024 · 84a3f3a · 84a3f3a
1 parent 3fbd1f5
commit 84a3f3a
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/schema.json b/schema.json
@@ -51,15 +51,15 @@
       },
       "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",
         "enum": ["chat", "instruct", "code_completion", "other"]
       },
       "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"