diff --git a/src/llamafactory/data/template.py b/src/llamafactory/data/template.py index cd3358f0..58393563 100644 --- a/src/llamafactory/data/template.py +++ b/src/llamafactory/data/template.py @@ -612,6 +612,16 @@ register_template( ) +register_template( + name="bailing", + format_user=StringFormatter(slots=["HUMAN{{content}}ASSISTANT"]), + format_system=StringFormatter(slots=["SYSTEM{{content}}"]), + format_observation=StringFormatter(slots=["OBSERVATION{{content}}ASSISTANT"]), + stop_words=["<|endoftext|>"], + efficient_eos=True, +) + + register_template( name="belle", format_user=StringFormatter(slots=["Human: {{content}}\n\nBelle: "]),