Tonic commited on
Commit
bd9f9cd
·
1 Parent(s): f9e3aeb

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +2 -2
app.py CHANGED
@@ -37,8 +37,8 @@ class OrcaChatBot:
37
  for role, message in self.conversation_history:
38
  if message.strip():
39
  prompt += f"<|im_start|>{role}\n{message}<|im_end|>\n"
40
- if role == "assistant":
41
- prompt += f"{self.tokenizer.eos_token}\n" # Add EOS token after assistant's messages
42
  prompt += "<|im_start|> assistant\n"
43
  return prompt
44
 
 
37
  for role, message in self.conversation_history:
38
  if message.strip():
39
  prompt += f"<|im_start|>{role}\n{message}<|im_end|>\n"
40
+ # if role == "assistant":
41
+ # prompt += f"<|im_end|>\n"
42
  prompt += "<|im_start|> assistant\n"
43
  return prompt
44