Fix repeat greeting (#808)

* fix repeated greeting * remove seperator between role and message
2023-10-16 01:52:21 +08:00 · 2023-10-16 01:52:21 +08:00 · a1ac199980
parent b50166500e
commit a1ac199980
1 changed files with 1 additions and 1 deletions
--- a/llama_cpp/llama_chat_format.py
+++ b/llama_cpp/llama_chat_format.py
@ -32,7 +32,7 @@ def _format_llama2(
    ret = system_message + sep
    for role, message in messages:
        if message:
-            ret += message + " "
+            ret += role + message + " "
        else:
            ret += role + " "
    return ret