build(docker): add chat template for chatglm (#165)

hyperonym · Apr 20, 2023 · aad86ad · aad86ad
1 parent b12e907
commit aad86ad
Showing 1 changed file with 17 additions and 0 deletions.
diff --git a/deployments/bundle/chatglm.chat.jinja b/deployments/bundle/chatglm.chat.jinja
@@ -0,0 +1,17 @@
+{% if messages|length >= 3 -%}
+{% for message in messages -%}
+{% if message.role == "system" -%}
+{{ message.content }}
+{% elif message.role == "user" -%}
+[Round {{ loop.index0 // 2 }}]
+问：{{ message.content }}
+{% elif message.role == "assistant" -%}
+答：{{ message.content }}
+{% endif -%}
+{% endfor -%}
+答：
+{%- else -%}
+{% for message in messages -%}
+{{ message.content }}
+{%- endfor -%}
+{% endif %}