diff --git a/matrix_pygmalion_bot/ai/runpod_pygmalion.py b/matrix_pygmalion_bot/ai/runpod_pygmalion.py index d67a01a..4137c21 100644 --- a/matrix_pygmalion_bot/ai/runpod_pygmalion.py +++ b/matrix_pygmalion_bot/ai/runpod_pygmalion.py @@ -175,8 +175,8 @@ async def generate_image(input_prompt: str, negative_prompt: str, api_url: str, "negative_prompt": negative_prompt, "width": 512, "height": 512, - "nsfw": True - } +# "nsfw": True + }, } logger.info(f"sending request to runpod.io") @@ -230,7 +230,7 @@ async def generate_image2(input_prompt: str, negative_prompt: str, api_key: str) return await generate_image(input_prompt, negative_prompt, "https://api.runpod.ai/v1/sd-openjourney/", api_key) async def generate_image3(input_prompt: str, negative_prompt: str, api_key: str): - return await generate_image(input_prompt, negative_prompt, "https://api.runpod.ai/v1/r9em5b11dkzvo0/", api_key) + return await generate_image(input_prompt, negative_prompt, "https://api.runpod.ai/v1/7me2asocq5lr01/", api_key) async def download_image(url, path): r = requests.get(url, stream=True) diff --git a/matrix_pygmalion_bot/core.py b/matrix_pygmalion_bot/core.py index 1c73650..4be049a 100644 --- a/matrix_pygmalion_bot/core.py +++ b/matrix_pygmalion_bot/core.py @@ -16,8 +16,8 @@ import json from .helpers import Event from .chatlog import BotChatHistory -#ai = importlib.import_module("matrix_pygmalion_bot.ai.runpod_pygmalion") -ai = importlib.import_module("matrix_pygmalion_bot.ai.stablehorde") +ai = importlib.import_module("matrix_pygmalion_bot.ai.runpod_pygmalion") +#ai = importlib.import_module("matrix_pygmalion_bot.ai.stablehorde") #from .llama_cpp import generate, get_full_prompt, get_full_prompt_chat_style #from .runpod_pygmalion import generate_sync, get_full_prompt import matrix_pygmalion_bot.translate as translate