Clemylia commited on
Commit
ff50dd1
·
verified ·
1 Parent(s): 3b15b61

Update chat_template.jinja

Browse files
Files changed (1) hide show
  1. chat_template.jinja +8 -2
chat_template.jinja CHANGED
@@ -1,4 +1,10 @@
1
  {%- for message in messages -%}
2
- {{ '### Instruction:\n' + message.content + '\n\n' }}
 
 
 
 
3
  {%- endfor -%}
4
- {{ '### Response:\n' }}
 
 
 
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 -%}