marclove commited on
Commit
c6ec0d5
1 Parent(s): dc2b319

Missing BOS token?

Browse files

I believe you're missing the very first `<s>` BOS token.

Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -35,7 +35,7 @@ examples=[
35
 
36
  def predict(message, chatbot):
37
 
38
- input_prompt = f"[INST] <<SYS>>\n{system_message}\n<</SYS>>\n\n "
39
  for interaction in chatbot:
40
  input_prompt = input_prompt + str(interaction[0]) + " [/INST] " + str(interaction[1]) + " </s><s> [INST] "
41
 
 
35
 
36
  def predict(message, chatbot):
37
 
38
+ input_prompt = f"<s>[INST] <<SYS>>\n{system_message}\n<</SYS>>\n\n "
39
  for interaction in chatbot:
40
  input_prompt = input_prompt + str(interaction[0]) + " [/INST] " + str(interaction[1]) + " </s><s> [INST] "
41