From 646a7885e7c721342b7f546ed717f1d4a68bc012 Mon Sep 17 00:00:00 2001 From: hoshi-hiyouga Date: Fri, 26 Apr 2024 02:20:47 +0800 Subject: [PATCH] delete llava template (use vicuna) --- src/llmtuner/data/template.py | 11 ----------- 1 file changed, 11 deletions(-) diff --git a/src/llmtuner/data/template.py b/src/llmtuner/data/template.py index 9a3673c3..73b22eb7 100644 --- a/src/llmtuner/data/template.py +++ b/src/llmtuner/data/template.py @@ -879,14 +879,3 @@ _register_template( format_user=StringFormatter(slots=[":{{content}}\n:"]), format_separator=EmptyFormatter(slots=["\n"]), ) - - -_register_template( - name="llava", - format_user=StringFormatter(slots=["USER: {{content}} ASSISTANT: "]), - format_assistant=StringFormatter(slots=["{{content}}", {"eos_token"}]), - default_system=( - "A chat between a curious user and an artificial intelligence assistant. " - "The assistant gives helpful, detailed, and polite answers to the user's questions." - ), -)