From 627f281e5b33326058734e42a8038722056d3a00 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: b3ac14ffc4c31cb54689c26b66963c20db64f21f --- 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:"]),