Przeglądaj źródła

make #15334 work without making copies of images

AUTOMATIC1111 1 rok temu
rodzic
commit
57727e554d
1 zmienionych plików z 1 dodań i 3 usunięć
  1. 1 3
      modules/postprocessing.py

+ 1 - 3
modules/postprocessing.py

@@ -66,7 +66,7 @@ def run_postprocessing(extras_mode, image, image_folder, input_dir, output_dir,
         if parameters:
             existing_pnginfo["parameters"] = parameters
 
-        initial_pp = scripts_postprocessing.PostprocessedImage(image_data.convert("RGBA")) if image_data.mode == "RGBA" else scripts_postprocessing.PostprocessedImage(image_data.convert("RGB"))
+        initial_pp = scripts_postprocessing.PostprocessedImage(image_data if image_data.mode in ("RGBA", "RGB") else image_data.convert("RGB"))
 
         scripts.scripts_postproc.run(initial_pp, args)
 
@@ -122,8 +122,6 @@ def run_postprocessing(extras_mode, image, image_folder, input_dir, output_dir,
             if extras_mode != 2 or show_extras_results:
                 outputs.append(pp.image)
 
-        image_data.close()
-
     devices.torch_gc()
     shared.state.end()
     return outputs, ui_common.plaintext_to_html(infotext), ''