diff --git a/matrix_pygmalion_bot/ai/runpod_pygmalion.py b/matrix_pygmalion_bot/ai/runpod_pygmalion.py index 9813486..38de21d 100644 --- a/matrix_pygmalion_bot/ai/runpod_pygmalion.py +++ b/matrix_pygmalion_bot/ai/runpod_pygmalion.py @@ -250,6 +250,14 @@ async def generate_image5(input_prompt: str, negative_prompt: str, api_key: str, async def generate_image6(input_prompt: str, negative_prompt: str, api_key: str, typing_fn): return await generate_image_automatic(input_prompt, negative_prompt, "https://api.runpod.ai/v1/5j1xzlsyw84vk5/", api_key, typing_fn) +async def generate_image7(input_prompt: str, negative_prompt: str, api_key: str, typing_fn): +# ChilloutMix + return await generate_image_automatic(input_prompt, negative_prompt, "https://api.runpod.ai/v2/rrjxafqx66osr4/", api_key, typing_fn) + +async def generate_image8(input_prompt: str, negative_prompt: str, api_key: str, typing_fn): + return await generate_image_automatic(input_prompt, negative_prompt, "https://api.runpod.ai/v2/vuyifmsasm3ix7/", api_key, typing_fn) + + async def serverless_automatic_request(payload, cmd, api_url: str, api_key: str, typing_fn): # Set the API endpoint URL endpoint = api_url + "run" diff --git a/matrix_pygmalion_bot/core.py b/matrix_pygmalion_bot/core.py index 60b5581..43d9889 100644 --- a/matrix_pygmalion_bot/core.py +++ b/matrix_pygmalion_bot/core.py @@ -123,6 +123,10 @@ class Callbacks(object): output = await ai.generate_image5(prompt, negative_prompt, self.bot.runpod_api_key, typing) elif num == 6: output = await ai.generate_image6(prompt, negative_prompt, self.bot.runpod_api_key, typing) + elif num == 7: + output = await ai.generate_image7(prompt, negative_prompt, self.bot.runpod_api_key, typing) + elif num == 8: + output = await ai.generate_image8(prompt, negative_prompt, self.bot.runpod_api_key, typing) else: raise ValueError('no image generator with that number') elif self.bot.service == "stablehorde":