From 6597b3df64178a8e4364f1b2867c1f2dab37c67c Mon Sep 17 00:00:00 2001 From: lllyasviel Date: Wed, 20 Sep 2023 06:00:52 -0700 Subject: [PATCH] fix bugradio (#455) --- modules/gradio_hijack.py | 7 ++++--- 1 file changed, 4 insertions(+), 3 deletions(-) diff --git a/modules/gradio_hijack.py b/modules/gradio_hijack.py index 119e9c3..9808f5d 100644 --- a/modules/gradio_hijack.py +++ b/modules/gradio_hijack.py @@ -265,10 +265,11 @@ class Image( if x is None: return x - mask = "" + mask = None + if self.tool == "sketch" and self.source in ["upload", "webcam"]: - assert isinstance(x, dict) - x, mask = x["image"], x["mask"] + if isinstance(x, dict): + x, mask = x["image"], x["mask"] assert isinstance(x, str) im = processing_utils.decode_base64_to_image(x)