Merge pull request #3715 from shwang95/master

Fix error caused by EXIF transpose when using custom scripts
3 years ago · bb21a4cb35
parent e6060a7e6b 5c864be010
commit bb21a4cb35
1 changed files with 2 additions and 1 deletions
--- a/modules/img2img.py
+++ b/modules/img2img.py
@ -81,7 +81,8 @@ def img2img(mode: int, prompt: str, negative_prompt: str, prompt_style: str, pro
        mask = None

    # Use the EXIF orientation of photos taken by smartphones.
-    image = ImageOps.exif_transpose(image) 
+    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]'