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

add Exaone3.0 template

Browse Source 
Former-commit-id: 3a9569647f5dbb1dcd9ef6e5cfc39ec8f9b41e89
This commit is contained in:
shing100 2024-09-30 09:17:41 +09:00
parent 8d20d6c95c
commit 0a633f8098
2 changed files with 21 additions and 0 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

12
src/llamafactory/data/template.py
View File

@ -644,6 +644,18 @@ _register_template(
) )
_register_template(
name="exaone",
format_user=StringFormatter(slots=["[|user|]{{content}}\n[|assistant|]"]),
format_system=StringFormatter(slots=["[|system|]{{content}}[|endofturn|]\n"]),
format_assistant=StringFormatter(slots=["{{content}}[|endofturn|]\n"]),
format_separator=EmptyFormatter(slots=["\n"]),
format_prefix=EmptyFormatter(slots=[""]),
stop_words=["[|endofturn|]"],
efficient_eos=True,
)
_register_template( _register_template(
name="falcon", name="falcon",
format_user=StringFormatter(slots=["User: {{content}}\nFalcon:"]), format_user=StringFormatter(slots=["User: {{content}}\nFalcon:"]),

9
src/llamafactory/extras/constants.py
View File

@ -471,6 +471,15 @@ register_model_group(
template="deepseekcoder", template="deepseekcoder",
) )
regiter_model_group(
models={
"EXAONE-3.0-7.8B-Instruct": {
DownloadSource.DEFAULT: "LGAI-EXAONE/EXAONE-3.0-7.8B-Instruct",
},
},
template="exaone",
)
register_model_group( register_model_group(
models={ models={