diff --git a/matrix_pygmalion_bot/ai/runpod_pygmalion.py b/matrix_pygmalion_bot/ai/runpod_pygmalion.py index d43523e..331fe6d 100644 --- a/matrix_pygmalion_bot/ai/runpod_pygmalion.py +++ b/matrix_pygmalion_bot/ai/runpod_pygmalion.py @@ -187,6 +187,7 @@ async def generate_image(input_prompt: str, negative_prompt: str, api_url: str, "negative_prompt": negative_prompt, "width": 512, "height": 768, + "num_outputs": 3, # "nsfw": True }, }