Version 0.3.6: add Phi and Minichat prompt templates

synw · Dec 27, 2023 · b12261e · b12261e
1 parent 3cad850
commit b12261e
Show file tree

Hide file tree

Showing 2 changed files with 19 additions and 9 deletions.
diff --git a/package.json b/package.json
@@ -1,6 +1,6 @@
 {
   "name": "modprompt",
-  "version": "0.3.5",
+  "version": "0.3.6",
   "description": "Prompt templates for language models",
   "license": "MIT",
   "scripts": {

diff --git a/src/db.ts b/src/db.ts
@@ -37,14 +37,6 @@ const templates: Record<string, LmTemplate> = {
     "prefix": "<s>",
     "stop": ["</s>"],
   },
-  "amazon": {
-    "id": "amazon",
-    "name": "Amazon",
-    "user": "<|prompter|>{prompt}</s>",
-    "assistant": "<|assistant|>",
-    "afterShot": "\n",
-    "stop": ["</s>"],
-  },
   "mistral": {
     "id": "mistral",
     "name": "Mistral",
@@ -183,6 +175,24 @@ const templates: Record<string, LmTemplate> = {
       "user": 2,
       "assistant": 1
     },
+  },
+  "minichat": {
+    "id": "minichat",
+    "name": "Minichat",
+    "user": "<s> [|User|] {prompt} </s>",
+    "assistant": "[|Assistant|]",
+    "stop": ["</s>", "[|User|]"],
+    "afterShot": "\n",
+  },
+  "phi": {
+    "id": "phi",
+    "name": "Phi",
+    "user": "Instruct: {prompt}",
+    "assistant": "Output:",
+    "linebreaks": {
+      "user": 1
+    },
+    "stop": ["</s>", "Instruct:"]
   }
 };