Fix error caused by EXIF transpose when using custom scripts

Some custom scripts read image directly and no need to select image in UI, this will cause error.
2022-10-26 20:32:55 +08:00 · 2022-10-26 20:32:55 +08:00 · 7bd8581e46
parent 99d728b5b1
commit 7bd8581e46
1 changed files with 2 additions and 1 deletions
--- a/modules/img2img.py
+++ b/modules/img2img.py
@ -80,6 +80,7 @@ def img2img(mode: int, prompt: str, negative_prompt: str, prompt_style: str, pro
        mask = None

    # Use the EXIF orientation of photos taken by smartphones.
+    if image is not None:
        image = ImageOps.exif_transpose(image) 

    assert 0. <= denoising_strength <= 1., 'can only work with strength in [0.0, 1.0]'