Merge pull request #5910 from Cuiyn/index

Support Index series models.
2025-12-16 11:50:35 +08:00 · 2024-11-02 20:16:54 +08:00
parent 83535bbe8b ecca9db66b
commit c58cc22d06
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|>:"]),