diff --git a/src/llmtuner/data/template.py b/src/llmtuner/data/template.py
index 3cd63881..459382bf 100644
--- 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=[