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

modify yi-vl template

Browse Source 
Former-commit-id: f113975b425e70bed2588ca55a2c62594fbf2283
This commit is contained in:
BUAADreamer 2024-05-14 16:45:28 +08:00
parent dd4619e9f3
commit 483ed64b43
1 changed files with 4 additions and 3 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

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

@ -857,15 +857,16 @@ _register_template(
_register_template( _register_template(
name="yi-vl", name="yivl",
format_user=StringFormatter(slots=["### Human:\n{{content}}\n### Assistant: "]), format_user=StringFormatter(slots=["### Human: {{content}}\n### Assistant:"]),
format_assistant=StringFormatter(slots=[" {{content}}"]),
stop_words=["###"], stop_words=["###"],
default_system=( default_system=(
"This is a chat between an inquisitive human and an AI assistant. " "This is a chat between an inquisitive human and an AI assistant. "
"Assume the role of the AI assistant. " "Assume the role of the AI assistant. "
"Read all the images carefully, and respond to the human's questions with informative, helpful, detailed and polite answers." "Read all the images carefully, and respond to the human's questions with informative, helpful, detailed and polite answers."
"这是一个好奇的人类和一个人工智能助手之间的对话。" "这是一个好奇的人类和一个人工智能助手之间的对话。"
"假设你扮演这个AI助手的角色。仔细阅读所有的图像，并对人类的问题做出信息丰富、有帮助、详细的和礼貌的回答。" "假设你扮演这个AI助手的角色。仔细阅读所有的图像，并对人类的问题做出信息丰富、有帮助、详细的和礼貌的回答。\n\n"
), ),
) )