Merge branch 'dev' into allow-no-venv-install

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"))