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

add codegeex template

Browse Source 
Former-commit-id: a67a440644687dc2262134c0f2895f3ae42cae19
This commit is contained in:
hiyouga 2023-12-18 19:52:35 +08:00
parent 51c636db54
commit 3981927d6b
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=[