From 4b27cf546078ba3798aa136af65eaca3f710fe7e Mon Sep 17 00:00:00 2001 From: hiyouga Date: Mon, 18 Dec 2023 19:52:35 +0800 Subject: [PATCH] add codegeex template Former-commit-id: a8222722b8097158f1c92e3729f41d411eff3926 --- src/llmtuner/data/template.py | 15 +++++++++++++++ 1 file changed, 15 insertions(+) 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=[