Browse Source

fix image api

master
Hendrik Langer 2 years ago
parent
commit
0f3032bd2e
  1. 2
      matrix_pygmalion_bot/ai/runpod_pygmalion.py
  2. 2
      matrix_pygmalion_bot/core.py

2
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 # Define your inputs
#payload.update({"api_endpoint": cmd}) payload.update({"api_endpoint": str(cmd)})
input_data = { input_data = {
"input": payload, "input": payload,
"cmd": cmd, "cmd": cmd,

2
matrix_pygmalion_bot/core.py

@ -271,6 +271,8 @@ class ChatBot(object):
self.chat_history = None self.chat_history = None
self.room_config = {} self.room_config = {}
self.negative_prompt = None
if STORE_PATH and not os.path.isdir(STORE_PATH): if STORE_PATH and not os.path.isdir(STORE_PATH):
os.mkdir(STORE_PATH) os.mkdir(STORE_PATH)

Loading…
Cancel
Save