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

Update chat_template.jinja

Browse files
Files changed (1) hide show
  1. chat_template.jinja +6 -6
chat_template.jinja CHANGED
@@ -1,10 +1,10 @@
1
  {%- for message in messages -%}
2
- {%- if loop.first and message.role == 'user' -%}
3
- {{ '<user> ' + message.content + ' ' }}
4
- {%- elif message.role == 'user' -%}
5
- {{ '<user> ' + message.content + ' ' }}
6
  {%- elif message.role == 'assistant' -%}
7
- {{ '<bot> ' + message.content + ' ' }}
8
  {%- endif -%}
9
  {%- endfor -%}
10
- {{ '<bot>' }}
 
 
 
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 -%}