support LLaMA-2

This commit is contained in:
hiyouga
2023-07-19 16:42:14 +08:00
parent 38eb1aaf55
commit 7a3ade8c69
3 changed files with 12 additions and 1 deletions

View File

@@ -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
)