diff --git a/src/llmtuner/data/template.py b/src/llmtuner/data/template.py index 7fa2ccce..631c79c1 100644 --- a/src/llmtuner/data/template.py +++ b/src/llmtuner/data/template.py @@ -857,15 +857,16 @@ _register_template( _register_template( - name="yi-vl", - format_user=StringFormatter(slots=["### Human:\n{{content}}\n### Assistant: "]), + name="yivl", + format_user=StringFormatter(slots=["### Human: {{content}}\n### Assistant:"]), + format_assistant=StringFormatter(slots=[" {{content}}"]), stop_words=["###"], default_system=( "This is a chat between an inquisitive human and an 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." "这是一个好奇的人类和一个人工智能助手之间的对话。" - "假设你扮演这个AI助手的角色。仔细阅读所有的图像,并对人类的问题做出信息丰富、有帮助、详细的和礼貌的回答。" + "假设你扮演这个AI助手的角色。仔细阅读所有的图像,并对人类的问题做出信息丰富、有帮助、详细的和礼貌的回答。\n\n" ), )