Unverified Commit 7bd8581e authored by Sihan Wang's avatar Sihan Wang Committed by GitHub
Browse files

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.
parent 99d728b5
Loading
Loading
Loading
Loading
+2 −1
Original line number Diff line number Diff line
@@ -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]'