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

merge data part to the text stream

Browse Source
This commit is contained in:
BUAADreamer
2024-04-25 19:58:47 +08:00
parent 2ee3046eb0
commit 42c90c8183
6 changed files with 18 additions and 38 deletions
Download Patch File Download Diff File Expand all files Collapse all files

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

@@ -1012,8 +1012,8 @@ _register_template(
_register_template(
name="llava",
format_user=StringFormatter(slots=["USER: {{content}} "]),
format_assistant=StringFormatter(slots=["ASSISTANT: {{content}}"]),
format_user=StringFormatter(slots=["USER: {{content}} ASSISTANT: "]),
format_assistant=StringFormatter(slots=["{{content}}", {"eos_token"}]),
default_system=(
"A chat between a curious user and an artificial intelligence assistant. "
"The assistant gives helpful, detailed, and polite answers to the user's questions."