diff options
| author | AUTOMATIC <16777216c@gmail.com> | 2022-09-12 15:59:53 +0000 |
|---|---|---|
| committer | AUTOMATIC <16777216c@gmail.com> | 2022-09-12 15:59:53 +0000 |
| commit | 482a6ce8cbdde82fecfabd4a47a7720b54676a54 (patch) | |
| tree | c93839e5b5ef94bcdfb6028b34b73766aba02887 /modules/extras.py | |
| parent | 843b2b64fcd41be4a9e934ba83a3a499c7aff5c0 (diff) | |
| download | stable-diffusion-webui-gfx803-482a6ce8cbdde82fecfabd4a47a7720b54676a54.tar.gz stable-diffusion-webui-gfx803-482a6ce8cbdde82fecfabd4a47a7720b54676a54.tar.bz2 stable-diffusion-webui-gfx803-482a6ce8cbdde82fecfabd4a47a7720b54676a54.zip | |
[Feature Request] Save defaults for extras & keep image parameters after using extras #251
Diffstat (limited to 'modules/extras.py')
| -rw-r--r-- | modules/extras.py | 4 |
1 files changed, 3 insertions, 1 deletions
diff --git a/modules/extras.py b/modules/extras.py index 40935f98..596cd172 100644 --- a/modules/extras.py +++ b/modules/extras.py @@ -13,6 +13,8 @@ cached_images = {} def run_extras(image, gfpgan_visibility, codeformer_visibility, codeformer_weight, upscaling_resize, extras_upscaler_1, extras_upscaler_2, extras_upscaler_2_visibility):
devices.torch_gc()
+ existing_pnginfo = image.info or {}
+
image = image.convert("RGB")
info = ""
@@ -65,7 +67,7 @@ def run_extras(image, gfpgan_visibility, codeformer_visibility, codeformer_weigh while len(cached_images) > 2:
del cached_images[next(iter(cached_images.keys()))]
- images.save_image(image, outpath, "", None, info=info, extension=opts.samples_format, short_filename=True, no_prompt=True, pnginfo_section_name="extras")
+ images.save_image(image, outpath, "", None, info=info, extension=opts.samples_format, short_filename=True, no_prompt=True, pnginfo_section_name="extras", existing_info=existing_pnginfo)
return image, plaintext_to_html(info), ''
|