Add the Deepseek v2 template

synw · Jun 25, 2024 · fe10473 · fe10473
1 parent aa517a9
commit fe10473
Show file tree

Hide file tree

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

diff --git a/src/db.ts b/src/db.ts
@@ -251,6 +251,20 @@ const templates: Record<string, LmTemplate> = {
     },
     "stop": ["<|EOT|>", "### Instruction:"]
   },
+  "deepseek2": {
+    "id": "deepseek2",
+    "name": "Deepseek v2",
+    "system": {
+      "schema": "<｜begin▁of▁sentence｜>{system}",
+    },
+    "user": "User: {prompt}",
+    "assistant": "Assistant:",
+    "linebreaks": {
+      "user": 2,
+      "system": 2,
+    },
+    "stop": ["<｜end▁of▁sentence｜>"]
+  },
   "opencodeinterpreter": {
     "id": "opencodeinterpreter",
     "name": "Open code interpreter",