diff --git a/backend-python/convert_safetensors.py b/backend-python/convert_safetensors.py index 876c4b5..c5a702b 100644 --- a/backend-python/convert_safetensors.py +++ b/backend-python/convert_safetensors.py @@ -102,6 +102,7 @@ if __name__ == "__main__": "time_mix_w2", "time_decay_w1", "time_decay_w2", + "lora.0", ], ) print(f"Saved to {args.output}") diff --git a/backend-python/routes/completion.py b/backend-python/routes/completion.py index 59d48f5..47e4dea 100644 --- a/backend-python/routes/completion.py +++ b/backend-python/routes/completion.py @@ -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")