add codegeex template

Former-commit-id: a67a440644687dc2262134c0f2895f3ae42cae19
2025-08-04 04:32:50 +08:00 · 2023-12-18 19:52:35 +08:00 · 2023-12-18 19:52:35 +08:00 · 3981927d6b
commit 3981927d6b
parent 51c636db54
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=[