Browse Source

readability

master
Hendrik Langer 2 years ago
parent
commit
73ab648bbc
  1. 2
      matrix_pygmalion_bot/ai/runpod.py

2
matrix_pygmalion_bot/ai/runpod.py

@ -115,7 +115,7 @@ async def generate_sync(
output.sort(key=len, reverse=True) output.sort(key=len, reverse=True)
text = output[0] text = output[0]
else: else:
text = r_json["output"] text = output
if api_mode == "runpod": if api_mode == "runpod":
reply = text.removeprefix(prompt) reply = text.removeprefix(prompt)
else: else:

Loading…
Cancel
Save