Spaces:

theodotus
/

llama-uk

Sleeping

theodotus commited on Aug 11, 2023

Commit

a268dd1

1 Parent(s): aa251d2

Fixed prompt

Files changed (1) hide show

app.py CHANGED Viewed

@@ -6,9 +6,9 @@ import gradio as gr
 def generate_prompt(history):
-    prompt = ""
     for chain in history[-2:-1]:
-        prompt += f"<s> <human>: {chain[0]}\n<bot>: {chain[1]}{end_token}\n"
     prompt += f"<human>: {history[-1][0]}\n<bot>:"
     return prompt

 def generate_prompt(history):
+    prompt = "<s> "
     for chain in history[-2:-1]:
+        prompt += f"<human>: {chain[0]}\n<bot>: {chain[1]}{end_token}\n"
     prompt += f"<human>: {history[-1][0]}\n<bot>:"
     return prompt