From 3d65c4ceabcfb9b660324e02f5f1450a729a4832 Mon Sep 17 00:00:00 2001 From: hoshi-hiyouga Date: Wed, 15 May 2024 14:20:39 +0800 Subject: [PATCH] Update template.py Former-commit-id: 780ca8306b31d5ac856f68de3abed7e838848464 --- src/llmtuner/data/template.py | 14 +++++++------- 1 file changed, 7 insertions(+), 7 deletions(-) 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=["###"], )