From 3981927d6b5999d1e7ffc375b79e6095c54d308c 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: a67a440644687dc2262134c0f2895f3ae42cae19 --- 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=[