add models

Former-commit-id: 758ae7937a41a95016e70180fb343011763c1b67
2023-12-06 13:33:18 +08:00
parent 7b7bfea37d
commit 24ce319b6f
2 changed files with 11 additions and 0 deletions
--- a/src/llmtuner/data/template.py
+++ b/src/llmtuner/data/template.py
@@ -724,6 +724,9 @@ register_template(
    sep=[
        "<|im_end|>\n"
    ],
+    stop_words=[
+        "<|im_end|>"
+    ],
    efficient_eos=True
 )