Skip to content

Commit 253a980

Browse files
authored
Add chat templates for ChatGLM (#3418)
1 parent 21539e6 commit 253a980

File tree

2 files changed

+36
-0
lines changed

2 files changed

+36
-0
lines changed

examples/template_chatglm.jinja

Lines changed: 18 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -0,0 +1,18 @@
1+
{%- set counter = namespace(index=0) -%}
2+
{%- for message in messages -%}
3+
{%- if message['role'] == 'user' -%}
4+
{{- '[Round ' + counter.index|string + ']\n问:' + message['content'] -}}
5+
{%- set counter.index = counter.index + 1 -%}
6+
{%- endif -%}
7+
{%- if message['role'] == 'assistant' -%}
8+
{{- '\n答:' + message['content'] -}}
9+
{%- if (loop.last and add_generation_prompt) or not loop.last -%}
10+
{{- '\n' -}}
11+
{%- endif -%}
12+
{%- endif -%}
13+
{%- endfor -%}
14+
15+
16+
{%- if add_generation_prompt and messages[-1]['role'] != 'assistant' -%}
17+
{{- '\n答:' -}}
18+
{%- endif -%}

examples/template_chatglm2.jinja

Lines changed: 18 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -0,0 +1,18 @@
1+
{%- set counter = namespace(index=1) -%}
2+
{%- for message in messages -%}
3+
{%- if message['role'] == 'user' -%}
4+
{{- '[Round ' + counter.index|string + ']\n\n问:' + message['content'] -}}
5+
{%- set counter.index = counter.index + 1 -%}
6+
{%- endif -%}
7+
{%- if message['role'] == 'assistant' -%}
8+
{{- '\n\n答:' + message['content'] -}}
9+
{%- if (loop.last and add_generation_prompt) or not loop.last -%}
10+
{{- '\n\n' -}}
11+
{%- endif -%}
12+
{%- endif -%}
13+
{%- endfor -%}
14+
15+
16+
{%- if add_generation_prompt and messages[-1]['role'] != 'assistant' -%}
17+
{{- '\n\n答:' -}}
18+
{%- endif -%}

0 commit comments

Comments
 (0)