Add a space in the Llama 2 prompt to match the reference implementation

OpenNMT · Sep 14, 2023 · e2b0133 · e2b0133
1 parent f9f584b
commit e2b0133
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/examples/llama2/chat.py b/examples/llama2/chat.py
@@ -110,7 +110,7 @@ def build_prompt(sp, dialog):
         [
             ["<s>"]
             + sp.encode_as_pieces(
-                f"{B_INST} {(prompt['content']).strip()} {E_INST} {(answer['content']).strip()}"
+                f"{B_INST} {(prompt['content']).strip()} {E_INST} {(answer['content']).strip()} "
             )
             + ["</s>"]
             for prompt, answer in zip(