rwitz commited on
Commit
e5dfca9
·
1 Parent(s): 3b0b13f

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +2 -2
app.py CHANGED
@@ -30,8 +30,8 @@ def write_elo_ratings(elo_ratings):
30
  def format_chatml_prompt(state):
31
  chatml_prompt = ""
32
  for message in state["history"]:
33
- chatml_prompt += "<|im_start|>"+message['role']+": "+ message['content']+"<|im_end|>\n"
34
- return chatml_prompt+"<|im_start|>Assistant:"
35
  def get_bot_response(url, prompt,state):
36
  chatml_prompt = format_chatml_prompt(state)
37
  payload = {
 
30
  def format_chatml_prompt(state):
31
  chatml_prompt = ""
32
  for message in state["history"]:
33
+ chatml_prompt += "<s>"+message['role']+": "+ message['content']+"</s>\n"
34
+ return chatml_prompt+"<s>Assistant:"
35
  def get_bot_response(url, prompt,state):
36
  chatml_prompt = format_chatml_prompt(state)
37
  payload = {