OrionStarAI
/

Orion-14B-LongChat

Text Generation

Model card Files Files and versions Community

chenxingphh commited on Jan 22

Commit

0d8b9ca

•

1 Parent(s): 9989b10

Update generation_utils.py

Files changed (1) hide show

generation_utils.py +1 -1

generation_utils.py CHANGED Viewed

@@ -10,7 +10,7 @@ def build_chat_input(tokenizer, messages: List[dict]):
         if message is None :
             continue
         if role == "user":
-            prompt += "Human: " + message + "\n\nAssistant: "
         if role == "assistant":
             prompt += message + "</s>"

         if message is None :
             continue
         if role == "user":
+            prompt += "Human: " + message + "\nAssistant: "
         if role == "assistant":
             prompt += message + "</s>"