Spaces:

theodotus
/

llama-uk

Sleeping

theodotus commited on Aug 11, 2023

Commit

cf610a7

1 Parent(s): 4aa20fd

History = 2

Files changed (1) hide show

app.py CHANGED Viewed

@@ -7,7 +7,7 @@ import gradio as gr
 def generate_prompt(history):
     prompt = ""
-    for chain in history[-3:-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 = ""
+    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