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

fix aquila template, repair sft packing mechanism

Browse Source
This commit is contained in:
hiyouga
2023-10-10 18:49:55 +08:00
parent e1dcb8e4dc
commit be420e4179
2 changed files with 18 additions and 7 deletions
Download Patch File Download Diff File Expand all files Collapse all files

8
src/llmtuner/extras/template.py
View File

@@ -423,7 +423,7 @@ register_template(
r"""
Supports: https://huggingface.co/qhduan/aquilachat-7b
Supports: https://huggingface.co/BAAI/AquilaChat-7B
"""
register_template(
name="aquila",
@@ -439,7 +439,11 @@ register_template(
),
sep=[
"###"
]
],
stop_words=[
"</s>"
],
efficient_eos=True
)