burtenshaw HF Staff Xenova HF Staff commited on
Commit
095599e
·
verified ·
1 Parent(s): daa5831

Create chat template file (#5)

Browse files

- Create chat template file (8998cdf6d5fe074d2886f747f1d69fac9e198d3b)
- simplify (5cb28c9b8367f8985b2f2d7deaef5ebbf0579109)


Co-authored-by: Joshua <[email protected]>

Files changed (1) hide show
  1. chat_template.jinja +15 -0
chat_template.jinja ADDED
@@ -0,0 +1,15 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ {{- "<|bos|>" -}}
2
+ {%- if messages[0]["role"] == "system" -%}
3
+ {{- "<|user_start|>" + messages[0]["content"] + "\n\n" + messages[1]["content"] + "<|user_end|>" -}}
4
+ {%- set messages = messages[2:] -%}
5
+ {%- endif -%}
6
+ {%- for message in messages -%}
7
+ {%- if loop.index0 % 2 == 0 -%}
8
+ {{- "<|user_start|>" + message["content"] + "<|user_end|>" -}}
9
+ {%- else -%}
10
+ {{- "<|assistant_start|>" + message["content"] + "<|assistant_end|>" -}}
11
+ {%- endif -%}
12
+ {%- endfor -%}
13
+ {%- if add_generation_prompt -%}
14
+ {{- "<|assistant_start|>" -}}
15
+ {%- endif -%}