chat completions forced append stop sequences if already existing

This commit is contained in:
Concedo 2024-04-10 19:25:30 +08:00
parent df596aeef3
commit d8fb97211f

View file

@ -744,6 +744,9 @@ class ServerRequestHandler(http.server.SimpleHTTPRequestHandler):
genparams["images"] = images_added
if len(genparams.get('stop_sequence', []))==0: #only set stop seq if it wont overwrite existing
genparams["stop_sequence"] = [user_message_start.strip(),assistant_message_start.strip()]
else:
genparams["stop_sequence"].append(user_message_start.strip())
genparams["stop_sequence"].append(assistant_message_start.strip())
genparams["trim_stop"] = True
elif api_format==5: