From 116fc90e9a74946f41be2ebf7510213834ba410d Mon Sep 17 00:00:00 2001 From: Yazan Agha-Schrader Date: Tue, 28 Nov 2023 07:17:45 +0100 Subject: [PATCH] Update promptFormats.js --- examples/server/public/promptFormats.js | 24 ++++++++++++++++++------ 1 file changed, 18 insertions(+), 6 deletions(-) diff --git a/examples/server/public/promptFormats.js b/examples/server/public/promptFormats.js index 6bbecd60f..f6b179a81 100644 --- a/examples/server/public/promptFormats.js +++ b/examples/server/public/promptFormats.js @@ -12,6 +12,12 @@ export const promptFormats = { char: "Response", user: "Instruction" }, +"bakllava": { + template: "{{history}}{{char}}:", + historyTemplate: "{{name}}: {{message}}\n", + char: "ASSISTANT", + user: "USER" +}, "chatml": { template: "<|im_start|>system\n{{prompt}}<|im_end|>\n{{history}}\n<|im_start|>{{char}}", historyTemplate: "<|im_start|>{{user}}\n{{message}}<|im_end|>", @@ -54,6 +60,18 @@ export const promptFormats = { char: "ASSISTANT", user: "USER" }, +"llama2": { + template: "[INST] <>\n{{prompt}}\n<>\n\n{{history}} [/INST] {{char}} [INST] ", + historyTemplate: "{{name}}: {{message}} [/INST]", + char: "llama", + user: "user" +}, +"llava": { + template: "{{history}}{{char}}:", + historyTemplate: "{{name}}: {{message}}\n", + char: "ASSISTANT", + user: "USER" +}, "leoHessianai": { template: "<|im_start|>system\n{{prompt}}<|im_end|>\n{{history}}\n<|im_start|>{{char}}", historyTemplate: "<|im_start|>{{user}}\n{{message}}<|im_end|>", @@ -66,12 +84,6 @@ export const promptFormats = { char: "ASSISTANT", user: "USER" }, -"llama2": { - template: "[INST] <>\n{{prompt}}\n<>\n\n{{history}} [/INST] {{char}} [INST] ", - historyTemplate: "{{name}}: {{message}} [/INST]", - char: "llama", - user: "user" -}, "marx": { template: "{{history}}\n{{char}}:", historyTemplate: "{{name}}: {{message}}",