Add the Gemma template

synw · Aug 1, 2024 · 811c46b · 811c46b
1 parent fe10473
commit 811c46b
Showing 1 changed file with 8 additions and 0 deletions.
diff --git a/src/db.ts b/src/db.ts
@@ -335,6 +335,14 @@ const templates: Record<string, LmTemplate> = {
       "schema": "<<SYS>>\n{system_prompt}\n<</SYS>>",
     },
   },
+  "gemma": {
+    "id": "gemma",
+    "name": "Gemma",
+    "user": "<start_of_turn>user\n{prompt}",
+    "assistant": "<end_of_turn>\n<start_of_turn>model",
+    "stop": ["<end_of_turn>"],
+    "afterShot": "\n"
+  },
 };
 
 export { templates }