423A35C7/LLaMA-Factory
1
0
Fork 0
mirror of https://github.com/hiyouga/LLaMA-Factory.git synced 2025-10-16 00:28:10 +08:00
Code Issues Packages Projects Releases Wiki Activity

add codegeex template

Browse Source 
Former-commit-id: a8222722b8097158f1c92e3729f41d411eff3926
This commit is contained in:
hiyouga 2023-12-18 19:52:35 +08:00
parent a53b2a643f
commit 4b27cf5460
1 changed files with 15 additions and 0 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

15
src/llmtuner/data/template.py
View File

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