fix for control images (#2011)

lllyasviel · Oct 8, 2024 · 7eb824a · 7eb824a
1 parent 4b69551
commit 7eb824a
Showing 1 changed file with 4 additions and 2 deletions.
diff --git a/modules_forge/forge_canvas/canvas.py b/modules_forge/forge_canvas/canvas.py
@@ -96,15 +96,17 @@ def preprocess(self, payload):
             return None
 
         image = base64_to_image(payload, numpy=self.numpy)
-        image.info = self.infotext
+        if hasattr(image, 'info'):
+            image.info = self.infotext
 
         return image
 
     def postprocess(self, value):
         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)