chenxingphh commited on
Commit
0d8b9ca
1 Parent(s): 9989b10

Update generation_utils.py

Browse files
Files changed (1) hide show
  1. generation_utils.py +1 -1
generation_utils.py CHANGED
@@ -10,7 +10,7 @@ def build_chat_input(tokenizer, messages: List[dict]):
10
  if message is None :
11
  continue
12
  if role == "user":
13
- prompt += "Human: " + message + "\n\nAssistant: "
14
  if role == "assistant":
15
  prompt += message + "</s>"
16
 
 
10
  if message is None :
11
  continue
12
  if role == "user":
13
+ prompt += "Human: " + message + "\nAssistant: "
14
  if role == "assistant":
15
  prompt += message + "</s>"
16