From ed5c641e8bba8a8e7c527cab97d21cc248f15230 Mon Sep 17 00:00:00 2001 From: "huangpan.foo" Date: Sat, 21 Sep 2024 19:33:30 +0800 Subject: [PATCH] Add deepseek-v2.5 template Former-commit-id: e80c1fe798fb2e076c0891a64300f1b6710176b6 --- src/llamafactory/data/template.py | 8 ++++++++ 1 file changed, 8 insertions(+) diff --git a/src/llamafactory/data/template.py b/src/llamafactory/data/template.py index 54da4757..01985801 100644 --- a/src/llamafactory/data/template.py +++ b/src/llamafactory/data/template.py @@ -611,6 +611,14 @@ _register_template( ) +_register_template( + name="deepseek_v2.5", + format_user=StringFormatter(slots=["<|User|>{{content}}<|Assistant|>"]), + format_system=StringFormatter(slots=["{{content}}"]), + format_prefix=EmptyFormatter(slots=[{"bos_token"}]), +) + + _register_template( name="deepseekcoder", format_user=StringFormatter(slots=["### Instruction:\n{{content}}\n### Response:"]),