aboutsummaryrefslogtreecommitdiff
path: root/modules/img2img.py
diff options
context:
space:
mode:
authorBruno Seoane <brunoseoaneamarillo@gmail.com>2022-11-04 16:40:13 -0300
committerBruno Seoane <brunoseoaneamarillo@gmail.com>2022-11-04 16:40:13 -0300
commitfd66f669ea25bad1409aec87ef14b8417009bddc (patch)
tree42348fb16d018920f6d7fc5970f6c54803ba5e72 /modules/img2img.py
parent31db25ecc8d9c3996e7bac00cc616ee12557b7d3 (diff)
parent89722fb5e4eda2adc5d3a6abf8babf8a58e80d69 (diff)
Merge branch 'master' of https://github.com/AUTOMATIC1111/stable-diffusion-webui
Diffstat (limited to 'modules/img2img.py')
-rw-r--r--modules/img2img.py5
1 files changed, 4 insertions, 1 deletions
diff --git a/modules/img2img.py b/modules/img2img.py
index 35c5df9b..be9f3653 100644
--- 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]'
@@ -137,6 +138,8 @@ def img2img(mode: int, prompt: str, negative_prompt: str, prompt_style: str, pro
if processed is None:
processed = process_images(p)
+ p.close()
+
shared.total_tqdm.clear()
generation_info_js = processed.js()