add TeleChat-12B/TeleChat-12B-v2 models

Former-commit-id: 0dd632fe9e
2025-12-15 03:10:35 +08:00 · 2024-05-29 15:00:37 +08:00
parent 2540ce58c0
commit ce1be3da4b
2 changed files with 26 additions and 0 deletions
--- a/src/llamafactory/data/template.py
+++ b/src/llamafactory/data/template.py
@@ -918,3 +918,13 @@ _register_template(
    format_user=StringFormatter(slots=["<human>:{{content}}\n<bot>:"]),
    format_separator=EmptyFormatter(slots=["\n"]),
 )
+
+
+_register_template(
+    name="telechat",
+    format_user=StringFormatter(slots=["<_user>{{content}}<_bot>"]),
+    format_system=StringFormatter(slots=["<_system>{{content}}<_end>"]),
+    default_system="",
+    stop_words=["<_end>"],
+    replace_eos=True,
+)