diff --git a/src/llmtuner/data/template.py b/src/llmtuner/data/template.py index 0b2ca0e6..b20c9203 100644 --- a/src/llmtuner/data/template.py +++ b/src/llmtuner/data/template.py @@ -857,17 +857,17 @@ _register_template( _register_template( - name="yivl", + name="yi_vl", format_user=StringFormatter(slots=["### Human: {{content}}\n### Assistant:"]), - format_assistant=StringFormatter(slots=[" {{content}}\n"]), - stop_words=["###"], + format_separator=EmptyFormatter(slots=["\n"]), 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助手的角色。仔细阅读所有的图像,并对人类的问题做出信息丰富、有帮助、详细的和礼貌的回答。\n\n" + "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助手的角色。" + "仔细阅读所有的图像,并对人类的问题做出信息丰富、有帮助、详细的和礼貌的回答。\n" ), + stop_words=["###"], )