From ce4f653121445e1c5ea906346f6f8e6752788621 Mon Sep 17 00:00:00 2001 From: hiyouga Date: Tue, 16 Apr 2024 03:10:02 +0800 Subject: [PATCH] add empty template Former-commit-id: a325ffa8a668bec354d2636683806acef105e196 --- src/llmtuner/data/template.py | 7 +++++++ 1 file changed, 7 insertions(+) diff --git a/src/llmtuner/data/template.py b/src/llmtuner/data/template.py index 7a1f4ab8..1311eda5 100644 --- a/src/llmtuner/data/template.py +++ b/src/llmtuner/data/template.py @@ -581,6 +581,13 @@ _register_template( ) +_register_template( + name="empty", + format_user=StringFormatter(slots=["{{content}}"]), + format_assistant=StringFormatter(slots=["{{content}}"]), +) + + _register_template( name="falcon", format_user=StringFormatter(slots=["User: {{content}}\nFalcon:"]),