hendrik
/
matrix-pygmalion-bot
1
0
Fork 0
Code Issues Pull Requests Projects Releases Wiki Activity
Browse Source

fix name replacements

master
Hendrik Langer 2 years ago
parent
7257b2cfa3
commit
8e9bbb1379
1 changed files with 3 additions and 3 deletions
Whitespace
Show all changes Ignore whitespace when comparing lines Ignore changes in amount of whitespace Ignore changes in whitespace at EOL
Unified View
Diff Options
Show Stats Download Patch File Download Diff File
  1. 6
      matrix_pygmalion_bot/ai/runpod_pygmalion.py

6
matrix_pygmalion_bot/ai/runpod_pygmalion.py
View File

@ -56,8 +56,8 @@ async def generate_sync(
reply = answer[:idx].strip() reply = answer[:idx].strip()
else: else:
reply = answer.removesuffix('<|endoftext|>').strip() reply = answer.removesuffix('<|endoftext|>').strip()
reply.replace("\n{bot_name}: ", " ") reply = reply.replace("\n{bot_name}: ", " ")
reply.replace("\n<BOT>: ", " ") reply = reply.replace("\n<BOT>: ", " ")
return reply return reply
elif status == 'IN_PROGRESS' or status == 'IN_QUEUE': elif status == 'IN_PROGRESS' or status == 'IN_QUEUE':
job_id = r_json["id"] job_id = r_json["id"]
@ -84,7 +84,7 @@ async def generate_sync(
else: else:
reply = answer.removesuffix('<|endoftext|>').strip() reply = answer.removesuffix('<|endoftext|>').strip()
reply = reply.replace("\n{bot_name}: ", " ") reply = reply.replace("\n{bot_name}: ", " ")
reply.replace("\n<BOT>: ", " ") reply = reply.replace("\n<BOT>: ", " ")
return reply return reply
else: else:
return "<ERROR>" return "<ERROR>"

Write Preview
Loading…
Cancel
Save