diff --git a/matrix_pygmalion_bot/ai/runpod_pygmalion.py b/matrix_pygmalion_bot/ai/runpod_pygmalion.py index b81217a..0a76e02 100644 --- a/matrix_pygmalion_bot/ai/runpod_pygmalion.py +++ b/matrix_pygmalion_bot/ai/runpod_pygmalion.py @@ -258,7 +258,14 @@ async def generate_image_automatic(input_prompt: str, negative_prompt: str, api_ input_data = { "input": { "prompt": input_prompt, - "steps": 5, + "steps": 20, + "restore_faces": True, + "gfpgan_visibility": 0.5, + "codeformer_visibility": 0.5, + "codeformer_weight": 0.5, + "override_settings": { + "filter_nsfw": False, + }, }, } diff --git a/matrix_pygmalion_bot/core.py b/matrix_pygmalion_bot/core.py index 4f77513..dc6daf1 100644 --- a/matrix_pygmalion_bot/core.py +++ b/matrix_pygmalion_bot/core.py @@ -145,7 +145,7 @@ class Callbacks(object): return elif event.body.startswith('!image4'): prompt = event.body.removeprefix('!image4').strip() - negative_prompt = "" + negative_prompt = "ugly, out of frame" if len(prompt) > 0: if self.bot.image_prompt: prompt.replace(self.bot.name, self.bot.image_prompt) diff --git a/requirements.txt b/requirements.txt index beaf7b4..0058daf 100644 --- a/requirements.txt +++ b/requirements.txt @@ -5,3 +5,4 @@ huggingface_hub python-magic pillow argostranslate +webuiapi