diff options
| author | Leon Feng <523684+leon0707@users.noreply.github.com> | 2023-07-18 04:24:14 -0400 |
|---|---|---|
| committer | GitHub <noreply@github.com> | 2023-07-18 04:24:14 -0400 |
| commit | a3730bd9becd2f1f5d209885b694b0dec178d110 (patch) | |
| tree | 8ac9948d89606f7519df786f07f6ddb93c3d2720 /modules/postprocessing.py | |
| parent | d6668347c8b85b11b696ac56777cc396e34ee1f9 (diff) | |
| parent | 871b8687a82bb2ca907d8a49c87aed7635b8fc33 (diff) | |
Merge branch 'dev' into fix-11805
Diffstat (limited to 'modules/postprocessing.py')
| -rw-r--r-- | modules/postprocessing.py | 7 |
1 files changed, 4 insertions, 3 deletions
diff --git a/modules/postprocessing.py b/modules/postprocessing.py index 736315e2..136e9c88 100644 --- a/modules/postprocessing.py +++ b/modules/postprocessing.py @@ -9,8 +9,7 @@ from modules.shared import opts def run_postprocessing(extras_mode, image, image_folder, input_dir, output_dir, show_extras_results, *args, save_output: bool = True):
devices.torch_gc()
- shared.state.begin()
- shared.state.job = 'extras'
+ shared.state.begin(job="extras")
image_data = []
image_names = []
@@ -54,7 +53,9 @@ def run_postprocessing(extras_mode, image, image_folder, input_dir, output_dir, for image, name in zip(image_data, image_names):
shared.state.textinfo = name
- existing_pnginfo = image.info or {}
+ parameters, existing_pnginfo = images.read_info_from_image(image)
+ if parameters:
+ existing_pnginfo["parameters"] = parameters
pp = scripts_postprocessing.PostprocessedImage(image.convert("RGB"))
|