support LLaMA-2

Former-commit-id: 04dfda054855ee9256586aacbd382f8fb0bfed04
2023-07-19 16:42:14 +08:00
parent 718f3382ad
commit 8b688251be
3 changed files with 12 additions and 1 deletions
--- a/src/llmtuner/extras/template.py
+++ b/src/llmtuner/extras/template.py
@@ -111,7 +111,7 @@ register_template(
           "explain why instead of answering something not correct. "
           "If you don't know the answer to a question, please don't share false information.\n<</SYS>>\n\n",
    prompt=" [INST] {query} [/INST] ",
-    sep="",
+    sep="</s>",
    use_history=True
 )