From 71b6b6ae47c51a0577be42d6ed7b263f827757a7 Mon Sep 17 00:00:00 2001
From: ljleb <lebel.louisjacob@gmail.com>
Date: Thu, 31 Aug 2023 00:53:07 -0400
Subject: [PATCH] dont need to await

---
 comfyui_custom_scripts/extensions/webuiNodes.js | 12 +++---------
 1 file changed, 3 insertions(+), 9 deletions(-)

diff --git a/comfyui_custom_scripts/extensions/webuiNodes.js b/comfyui_custom_scripts/extensions/webuiNodes.js
index dc4fa87..af97722 100644
--- a/comfyui_custom_scripts/extensions/webuiNodes.js
+++ b/comfyui_custom_scripts/extensions/webuiNodes.js
@@ -68,15 +68,6 @@ app.registerExtension({
             }
         }
     },
-    async setup() {
-        try {
-            await iframeRegisteredEvent;
-        } catch {
-            return;
-        }
-
-        app.loadGraphData();
-    },
     async nodeCreated(node) {
         let iframeInfo = null;
 
@@ -97,6 +88,9 @@ app.registerExtension({
         // 240 and 40 are empirical values that seem to work
         node.size = [240, 40 + distanceBetweenIoSlots * maxIoLength];
     },
+    async setup() {
+        app.loadGraphData();
+    },
 });
 
 const webuiIoNodeNames = [