From 3e0afbb3a24b995bb53c3fe2d046adcb91a679bc Mon Sep 17 00:00:00 2001
From: w-e-w <40751091+w-e-w@users.noreply.github.com>
Date: Fri, 5 Apr 2024 14:57:37 +0900
Subject: [PATCH] handle when image is None

---
 scripts/sd-image-editor.py | 2 ++
 1 file changed, 2 insertions(+)

diff --git a/scripts/sd-image-editor.py b/scripts/sd-image-editor.py
index 7548813..5f653d8 100644
--- a/scripts/sd-image-editor.py
+++ b/scripts/sd-image-editor.py
@@ -13,6 +13,8 @@ from PIL import Image, ImageEnhance, ImageFilter, ImageTransform
 save_path = os.path.join("output", "img2img-images", "sd-image-editor")
 
 def edit(img, degree, expand, flip, interpolate_mode, color, contrast, brightness, sharpness):
+    if img is None:
+        return None
     # Flip
     if flip:
         img = img.transpose(method=Image.Transpose.FLIP_LEFT_RIGHT)