Minor fix in deepseek2 template

- Updated package version to 0.7.6 in package.json - Added new stop token "<｜tool▁calls▁end｜>" to db.ts
synw · Aug 27, 2024 · 214e706 · 214e706
1 parent 811c46b
commit 214e706
Show file tree

Hide file tree

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

diff --git a/src/db.ts b/src/db.ts
@@ -263,7 +263,7 @@ const templates: Record<string, LmTemplate> = {
       "user": 2,
       "system": 2,
     },
-    "stop": ["<｜end▁of▁sentence｜>"]
+    "stop": ["<｜end▁of▁sentence｜>", "<｜tool▁calls▁end｜>"]
   },
   "opencodeinterpreter": {
     "id": "opencodeinterpreter",
@@ -311,7 +311,6 @@ const templates: Record<string, LmTemplate> = {
     "name": "Llama 3",
     "user": "<|start_header_id|>user<|end_header_id|>\n\n{prompt}<|eot_id|>",
     "assistant": "<|start_header_id|>assistant<|end_header_id|>",
-    "prefix": "<|begin_of_text|>",
     "stop": [
       "<|eot_id|>",
       "<|end_of_text|>"