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

modify yi-vl template

Browse Source 
Former-commit-id: d72e6f8dfd670533f3bbdf0bf5e7d596e2dd34ac
This commit is contained in:
BUAADreamer 2024-05-14 16:45:28 +08:00
parent 661565fc2e
commit 9e247245a2
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"
), ),
) )