small fix

This commit is contained in:
josc146 2024-04-30 21:52:47 +08:00
parent 2a0ad19bc5
commit 2d853f92b9
2 changed files with 2 additions and 1 deletions

View File

@ -102,6 +102,7 @@ if __name__ == "__main__":
"time_mix_w2",
"time_decay_w1",
"time_decay_w2",
"lora.0",
],
)
print(f"Saved to {args.output}")

View File

@ -382,7 +382,7 @@ async def chat_completions(body: ChatCompletionBody, request: Request):
body.stop.append(f"\n\n{user_code}")
body.stop.append(f"\n\n{bot_code}")
elif body.stop is None:
body.stop = default_stop
body.stop = default_stop + [f"\n\n{user_code}", f"\n\n{bot_code}"]
# if not body.presystem:
# body.stop.append("\n\n")