From 7e31721803fdbd4d91198d25e2cf06b132335e3a Mon Sep 17 00:00:00 2001
From: layerdiffusion <19834515+lllyasviel@users.noreply.github.com>
Date: Sat, 27 Jul 2024 07:50:59 -0700
Subject: [PATCH] fix #861

---
 modules_forge/forge_canvas/canvas.py | 8 +++++---
 1 file changed, 5 insertions(+), 3 deletions(-)

diff --git a/modules_forge/forge_canvas/canvas.py b/modules_forge/forge_canvas/canvas.py
index 39efd5c5..ff5fb5a3 100644
--- a/modules_forge/forge_canvas/canvas.py
+++ b/modules_forge/forge_canvas/canvas.py
@@ -10,9 +10,11 @@ create_or_modify_pyi_org = gradio.component_meta.create_or_modify_pyi
 
 
 def create_or_modify_pyi_org_patched(component_class, class_name, events):
-    if component_class.__name__ == 'LogicalImage':
-        return
-
+    try:
+        if component_class.__name__ == 'LogicalImage':
+            return
+    except:
+        pass
     return create_or_modify_pyi_org(component_class, class_name, events)