Fix adding system prompt in chat_template

#200
by mspanrin - opened

Currently the system message is only attached on the last message

{%- if loop.last and system_message is defined %}
  {{- "[INST] " + system_message + "\n\n" + message["content"] + "[/INST]" }}
{%- else %}
  {{- "[INST] " + message["content"] + "[/INST]" }}

however, system prompt should be added on the first user message

{%- if loop.first and system_message is defined %}
  {{- "[INST] " + system_message + "\n\n" + message["content"] + "[/INST]" }}
{%- else %}
  {{- "[INST] " + message["content"] + "[/INST]" }}

Sign up or log in to comment