add codegeex template

Former-commit-id: a8222722b8097158f1c92e3729f41d411eff3926
2025-10-14 23:58:11 +08:00 · 2023-12-18 19:52:35 +08:00 · 2023-12-18 19:52:35 +08:00 · 4b27cf5460
commit 4b27cf5460
parent a53b2a643f
1 changed files with 15 additions and 0 deletions
--- a/src/llmtuner/data/template.py
+++ b/src/llmtuner/data/template.py
@ -402,6 +402,21 @@ register_template(
 )


+register_template(
+    name="codegeex2",
+    prefix=[
+        {"token": "[gMASK]"},
+        {"token": "sop"},
+        "{{system}}"
+    ],
+    prompt=[
+        "{{query}}"
+    ],
+    system="",
+    sep=[]
+)
+
+
 register_template(
    name="deepseek",
    prefix=[