diff --git a/matrix_pygmalion_bot/ai/runpod_pygmalion.py b/matrix_pygmalion_bot/ai/runpod_pygmalion.py index 38de21d..6505d81 100644 --- a/matrix_pygmalion_bot/ai/runpod_pygmalion.py +++ b/matrix_pygmalion_bot/ai/runpod_pygmalion.py @@ -233,21 +233,27 @@ async def generate_image(input_prompt: str, negative_prompt: str, api_url: str, return files async def generate_image1(input_prompt: str, negative_prompt: str, api_key: str, typing_fn): +# AnythingV4 return await generate_image(input_prompt, negative_prompt, "https://api.runpod.ai/v1/sd-anything-v4/", api_key, typing_fn) async def generate_image2(input_prompt: str, negative_prompt: str, api_key: str, typing_fn): +# OpenJourney return await generate_image(input_prompt, negative_prompt, "https://api.runpod.ai/v1/sd-openjourney/", api_key, typing_fn) async def generate_image3(input_prompt: str, negative_prompt: str, api_key: str, typing_fn): +# Hassanblend return await generate_image(input_prompt, negative_prompt, "https://api.runpod.ai/v1/mf5f6mocy8bsvx/", api_key, typing_fn) async def generate_image4(input_prompt: str, negative_prompt: str, api_key: str, typing_fn): +# DeliberateV2 return await generate_image_automatic(input_prompt, negative_prompt, "https://api.runpod.ai/v1/lxdhmiccp3vdsf/", api_key, typing_fn) async def generate_image5(input_prompt: str, negative_prompt: str, api_key: str, typing_fn): +# Hassanblend return await generate_image_automatic(input_prompt, negative_prompt, "https://api.runpod.ai/v1/13rrs00l7yxikf/", api_key, typing_fn) async def generate_image6(input_prompt: str, negative_prompt: str, api_key: str, typing_fn): +# PFG 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): @@ -255,6 +261,7 @@ async def generate_image7(input_prompt: str, negative_prompt: str, api_key: str, 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): +# AbyssOrangeMixV3 return await generate_image_automatic(input_prompt, negative_prompt, "https://api.runpod.ai/v2/vuyifmsasm3ix7/", api_key, typing_fn)