Unverified Commit 1142a87c authored by AUTOMATIC1111's avatar AUTOMATIC1111 Committed by GitHub
Browse files

Merge pull request #9219 from Z-nonymous/master

Fix #9185
parents 579e13df 18e4ca46
Loading
Loading
Loading
Loading
+5 −2
Original line number Diff line number Diff line
@@ -4,7 +4,7 @@ import sys
import traceback

import numpy as np
from PIL import Image, ImageOps, ImageFilter, ImageEnhance, ImageChops
from PIL import Image, ImageOps, ImageFilter, ImageEnhance, ImageChops, UnidentifiedImageError

from modules import devices, sd_samplers
from modules.generation_parameters_copypaste import create_override_settings_dict
@@ -46,7 +46,10 @@ def process_batch(p, input_dir, output_dir, inpaint_mask_dir, args):
        if state.interrupted:
            break

        try:
            img = Image.open(image)
        except UnidentifiedImageError:
            continue
        # Use the EXIF orientation of photos taken by smartphones.
        img = ImageOps.exif_transpose(img)
        p.init_images = [img] * p.batch_size