diff --git a/matrix_pygmalion_bot/ai/runpod_pygmalion.py b/matrix_pygmalion_bot/ai/runpod_pygmalion.py index 0fd74cb..78ca0c7 100644 --- a/matrix_pygmalion_bot/ai/runpod_pygmalion.py +++ b/matrix_pygmalion_bot/ai/runpod_pygmalion.py @@ -76,8 +76,10 @@ async def generate_sync( reply = answer[:idx].strip() else: reply = answer.removesuffix('<|endoftext|>').strip() - reply = reply.replace("\n{bot.name}: ", " ") - reply = reply.replace("\n: ", " ") + reply = reply.replace("\\n{bot.name}: ", " ") + reply = reply.replace("\\n: ", " ") + reply = reply.replace("", "{bot.name}") + reply = reply.replace("", "You") return reply else: err_msg = r_json["error"] if "error" in r_json else ""