Merge pull request #9513 from ilya-khadykin/fix_batch_processing

fix(extras): fix batch image processing on 'Extras\Batch Process' tab

1 files changed, 4 insertions, 2 deletions

diff --git a/modules/postprocessing.py b/modules/postprocessing.py
index ff055aae..9cb80957 100644
--- a/modules/postprocessing.py
+++ b/modules/postprocessing.py
@@ -1,4 +1,6 @@
 import os

+import tempfile

+from typing import List

 

 from PIL import Image

 

@@ -6,7 +8,7 @@ from modules import shared, images, devices, scripts, scripts_postprocessing, ui
 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):

+def run_postprocessing(extras_mode, image, image_folder: List[tempfile.NamedTemporaryFile], input_dir, output_dir, show_extras_results, *args, save_output: bool = True):

     devices.torch_gc()

 

     shared.state.begin()

@@ -22,7 +24,7 @@ def run_postprocessing(extras_mode, image, image_folder, input_dir, output_dir,
                 image = img

                 fn = ''

             else:

-                image = Image.open(img)

+                image = Image.open(os.path.abspath(img.name))

                 fn = os.path.splitext(img.orig_name)[0]

 

             image_data.append(image)