aboutsummaryrefslogtreecommitdiffstats
path: root/modules/generation_parameters_copypaste.py
diff options
context:
space:
mode:
authorAUTOMATIC <16777216c@gmail.com>2023-06-04 07:56:48 +0000
committerAUTOMATIC <16777216c@gmail.com>2023-06-04 07:56:48 +0000
commitf98f4f73aa4898c754681f411608df5f248619f6 (patch)
tree6b8a1f63a622f373258a1c64ea5feb8751ceb071 /modules/generation_parameters_copypaste.py
parent30bbb8bce377e627be6c582315691358d9b611d8 (diff)
downloadstable-diffusion-webui-gfx803-f98f4f73aa4898c754681f411608df5f248619f6.tar.gz
stable-diffusion-webui-gfx803-f98f4f73aa4898c754681f411608df5f248619f6.tar.bz2
stable-diffusion-webui-gfx803-f98f4f73aa4898c754681f411608df5f248619f6.zip
infer styles from prompts, and an option to control the behavior
Diffstat (limited to 'modules/generation_parameters_copypaste.py')
-rw-r--r--modules/generation_parameters_copypaste.py8
1 files changed, 8 insertions, 0 deletions
diff --git a/modules/generation_parameters_copypaste.py b/modules/generation_parameters_copypaste.py
index 071bd9ea..4c420e5f 100644
--- a/modules/generation_parameters_copypaste.py
+++ b/modules/generation_parameters_copypaste.py
@@ -265,6 +265,14 @@ Steps: 20, Sampler: Euler a, CFG scale: 7, Seed: 965400086, Size: 512x512, Model
else:
prompt += ("" if prompt == "" else "\n") + line
+ if shared.opts.infotext_styles != "Ignore":
+ found_styles, prompt, negative_prompt = shared.prompt_styles.extract_styles_from_prompt(prompt, negative_prompt)
+
+ if shared.opts.infotext_styles == "Apply":
+ res["Styles array"] = found_styles
+ elif shared.opts.infotext_styles == "Apply if any" and found_styles:
+ res["Styles array"] = found_styles
+
res["Prompt"] = prompt
res["Negative prompt"] = negative_prompt