From 7eb824a0daf2efe9e8e036a55e19de247f3f5d4d Mon Sep 17 00:00:00 2001 From: DenOfEquity <166248528+DenOfEquity@users.noreply.github.com> Date: Tue, 8 Oct 2024 13:37:46 +0100 Subject: [PATCH] fix for control images (#2011) --- modules_forge/forge_canvas/canvas.py | 6 ++++-- 1 file changed, 4 insertions(+), 2 deletions(-) diff --git a/modules_forge/forge_canvas/canvas.py b/modules_forge/forge_canvas/canvas.py index b7e354af..a343f27e 100644 --- a/modules_forge/forge_canvas/canvas.py +++ b/modules_forge/forge_canvas/canvas.py @@ -96,7 +96,8 @@ class LogicalImage(gr.Textbox): return None image = base64_to_image(payload, numpy=self.numpy) - image.info = self.infotext + if hasattr(image, 'info'): + image.info = self.infotext return image @@ -104,7 +105,8 @@ class LogicalImage(gr.Textbox): if value is None: return None - self.infotext = value.info + if hasattr(value, 'info'): + self.infotext = value.info return image_to_base64(value, numpy=self.numpy)