somehumanperson1 commited on
Commit
7d4d6fa
1 Parent(s): ffbdf76

Update defaults/MODEL_PROMPT_TEMPLATE

Browse files
Files changed (1) hide show
  1. defaults/MODEL_PROMPT_TEMPLATE +1 -1
defaults/MODEL_PROMPT_TEMPLATE CHANGED
@@ -1 +1 @@
1
- <s>{{#each messages}}{{#ifUser}}[INST] {{#if @first}}{{#if @root.preprompt}}{{@root.preprompt}}\n{{/if}}{{/if}} {{content}} [/INST]{{/ifUser}}{{#ifAssistant}}{{content}}</s> {{/ifAssistant}}{{/each}}
 
1
+ {% for message in messages %}\n{% if message['role'] == 'user' %}\n{{ '<|user|>\n' + message['content'] + eos_token }}\n{% elif message['role'] == 'system' %}\n{{ '<|system|>\n' + message['content'] + eos_token }}\n{% elif message['role'] == 'assistant' %}\n{{ '<|assistant|>\n' + message['content'] + eos_token }}\n{% endif %}\n{% if loop.last and add_generation_prompt %}\n{{ '<|assistant|>' }}\n{% endif %}\n{% endfor %}