diff options
| author | AUTOMATIC1111 <16777216c@gmail.com> | 2024-01-20 07:24:59 +0000 |
|---|---|---|
| committer | GitHub <noreply@github.com> | 2024-01-20 07:24:59 +0000 |
| commit | 5df56b3980fed9efc4d0ba3a5a52782c34acd9f1 (patch) | |
| tree | 35bada6a0b1e218bdfdaaf4ed3c20c14a6661f93 /modules/infotext_utils.py | |
| parent | 9cdd161160a890f7993f27fd3616421eb665877b (diff) | |
| parent | 2cf23099ebb81832a27c8016f14062885f5a9c98 (diff) | |
| download | stable-diffusion-webui-gfx803-5df56b3980fed9efc4d0ba3a5a52782c34acd9f1.tar.gz stable-diffusion-webui-gfx803-5df56b3980fed9efc4d0ba3a5a52782c34acd9f1.tar.bz2 stable-diffusion-webui-gfx803-5df56b3980fed9efc4d0ba3a5a52782c34acd9f1.zip | |
Merge pull request #14626 from AUTOMATIC1111/hr-button-fix
more Hr button fix
Diffstat (limited to 'modules/infotext_utils.py')
| -rw-r--r-- | modules/infotext_utils.py | 8 |
1 files changed, 5 insertions, 3 deletions
diff --git a/modules/infotext_utils.py b/modules/infotext_utils.py index 9a02cdf2..1049c6c3 100644 --- a/modules/infotext_utils.py +++ b/modules/infotext_utils.py @@ -230,7 +230,7 @@ def restore_old_hires_fix_params(res): res['Hires resize-2'] = height
-def parse_generation_parameters(x: str):
+def parse_generation_parameters(x: str, skip_fields: list[str] | None = None):
"""parses generation parameters string, the one you see in text field under the picture in UI:
```
girl with an artist's beret, determined, blue eyes, desert scene, computer monitors, heavy makeup, by Alphonse Mucha and Charlie Bowater, ((eyeshadow)), (coquettish), detailed, intricate
@@ -240,6 +240,8 @@ Steps: 20, Sampler: Euler a, CFG scale: 7, Seed: 965400086, Size: 512x512, Model returns a dict with field values
"""
+ if skip_fields is None:
+ skip_fields = shared.opts.infotext_skip_pasting
res = {}
@@ -356,8 +358,8 @@ Steps: 20, Sampler: Euler a, CFG scale: 7, Seed: 965400086, Size: 512x512, Model infotext_versions.backcompat(res)
- skip = set(shared.opts.infotext_skip_pasting)
- res = {k: v for k, v in res.items() if k not in skip}
+ for key in skip_fields:
+ res.pop(key, None)
return res
|