Clemylia commited on
Commit
149513c
·
verified ·
1 Parent(s): f1d03ab

Update chat_template.jinja

Browse files
Files changed (1) hide show
  1. chat_template.jinja +4 -4
chat_template.jinja CHANGED
@@ -1,10 +1,10 @@
1
  {%- for message in messages -%}
2
  {%- if message.role == 'user' -%}
3
- {{ '### Instruction:\n' + message.content + '\n\n' }}
4
  {%- elif message.role == 'assistant' -%}
5
- {{ '### Response:\n' + message.content + ' <|end|>\n' }}
6
  {%- endif -%}
7
  {%- endfor -%}
8
- {%- if add_generation_prompt -%}
9
- {{ '### Response:\n' }}
10
  {%- endif -%}
 
1
  {%- for message in messages -%}
2
  {%- if message.role == 'user' -%}
3
+ {{ '<user> ' + message.content }}
4
  {%- elif message.role == 'assistant' -%}
5
+ {{ '<bot> ' + message.content }}
6
  {%- endif -%}
7
  {%- endfor -%}
8
+ {%- if messages[-1].role == 'user' -%}
9
+ {{ '<bot>' }}
10
  {%- endif -%}