Update promptFormats.js

This commit is contained in:
Yazan Agha-Schrader 2023-11-28 07:15:03 +01:00 committed by GitHub
parent 9dcb514b1d
commit 2e4c05e00a
No known key found for this signature in database
GPG key ID: 4AEE18F83AFDEB23

View file

@ -12,12 +12,6 @@ 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|>",
@ -60,18 +54,6 @@ export const promptFormats = {
char: "ASSISTANT",
user: "USER"
},
"llama2": {
template: "<s>[INST] <<SYS>>\n{{prompt}}\n<</SYS>>\n\n{{history}} [/INST] {{char}} </s><s>[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|>",
@ -84,6 +66,12 @@ export const promptFormats = {
char: "ASSISTANT",
user: "USER"
},
"llama2": {
template: "<s>[INST] <<SYS>>\n{{prompt}}\n<</SYS>>\n\n{{history}} [/INST] {{char}} </s><s>[INST] ",
historyTemplate: "{{name}}: {{message}} [/INST]",
char: "llama",
user: "user"
},
"marx": {
template: "{{history}}\n{{char}}:",
historyTemplate: "{{name}}: {{message}}",
@ -138,6 +126,12 @@ export const promptFormats = {
char: "Response",
user: "Input"
},
"openChat": {
template: "{{history}}{{char}}",
historyTemplate: "GPT4 {{user}}: {{message}}<|end_of_turn|>",
char: "Assistant",
user: "User"
},
"openhermes2Mistral": {
template: "<|im_start|>system\n{{prompt}}<|im_end|>\n{{history}}\n<|im_start|>{{char}}",
historyTemplate: "<|im_start|>{{user}}\n{{message}}<|im_end|>",
@ -174,6 +168,12 @@ export const promptFormats = {
char: "ASSISTANT",
user: "USER"
},
"starlingLM": {
template: "{{history}}{{char}}",
historyTemplate: "GPT4 {{user}}: {{message}}<|end_of_turn|>",
char: "Assistant",
user: "User"
},
"sydney": {
template: "{{prompt}}\n\n{{history}}\n{{char}}",
historyTemplate: "### {{name}}:\n{{message}}\n",