Merge pull request #5910 from Cuiyn/index

Support Index series models. Former-commit-id: c58cc22d06
2025-12-20 13:50:35 +08:00 · 2024-11-02 20:16:54 +08:00
parent 0f53217bbc 51177c933a
commit 6f79974e8b
4 changed files with 39 additions and 2 deletions
--- a/src/llamafactory/data/template.py
+++ b/src/llamafactory/data/template.py
@@ -691,6 +691,14 @@ _register_template(
 )


+_register_template(
+    name="index",
+    format_user=StringFormatter(slots=["reserved_0{{content}}reserved_1"]),
+    format_system=StringFormatter(slots=["<unk>{{content}}"]),
+    efficient_eos=True,
+)
+
+
 _register_template(
    name="intern",
    format_user=StringFormatter(slots=["<|User|>:{{content}}\n<|Bot|>:"]),