From 73ab648bbc33c7a01104661ca928457725c005b6 Mon Sep 17 00:00:00 2001 From: Hendrik Langer Date: Wed, 19 Apr 2023 18:35:16 +0200 Subject: [PATCH] readability --- matrix_pygmalion_bot/ai/runpod.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/matrix_pygmalion_bot/ai/runpod.py b/matrix_pygmalion_bot/ai/runpod.py index d7e5179..b34931a 100644 --- a/matrix_pygmalion_bot/ai/runpod.py +++ b/matrix_pygmalion_bot/ai/runpod.py @@ -115,7 +115,7 @@ async def generate_sync( output.sort(key=len, reverse=True) text = output[0] else: - text = r_json["output"] + text = output if api_mode == "runpod": reply = text.removeprefix(prompt) else: