From 0f3032bd2edd074259962da9977b219529c7c0a1 Mon Sep 17 00:00:00 2001 From: Hendrik Langer Date: Wed, 5 Apr 2023 23:25:29 +0200 Subject: [PATCH] fix image api --- matrix_pygmalion_bot/ai/runpod_pygmalion.py | 2 +- matrix_pygmalion_bot/core.py | 2 ++ 2 files changed, 3 insertions(+), 1 deletion(-) diff --git a/matrix_pygmalion_bot/ai/runpod_pygmalion.py b/matrix_pygmalion_bot/ai/runpod_pygmalion.py index 4c4d3c3..73ab64b 100644 --- a/matrix_pygmalion_bot/ai/runpod_pygmalion.py +++ b/matrix_pygmalion_bot/ai/runpod_pygmalion.py @@ -266,7 +266,7 @@ async def serverless_automatic_request(payload, cmd, api_url: str, api_key: str) } # Define your inputs - #payload.update({"api_endpoint": cmd}) + payload.update({"api_endpoint": str(cmd)}) input_data = { "input": payload, "cmd": cmd, diff --git a/matrix_pygmalion_bot/core.py b/matrix_pygmalion_bot/core.py index 01796c9..cafe8c9 100644 --- a/matrix_pygmalion_bot/core.py +++ b/matrix_pygmalion_bot/core.py @@ -271,6 +271,8 @@ class ChatBot(object): self.chat_history = None self.room_config = {} + self.negative_prompt = None + if STORE_PATH and not os.path.isdir(STORE_PATH): os.mkdir(STORE_PATH)