add models

2023-12-06 13:33:18 +08:00
parent d3dccd0693
commit e25f7bae16
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
 )