Reregister nodes when pressing refresh button.

comfyanonymous · Dec 31, 2023 · 36e15f2 · 36e15f2
1 parent 1b103e0
commit 36e15f2
Showing 1 changed file with 2 additions and 6 deletions.
diff --git a/web/scripts/app.js b/web/scripts/app.js
@@ -2020,12 +2020,8 @@ export class ComfyApp {
 	async refreshComboInNodes() {
 		const defs = await api.getNodeDefs();
 
-		for(const nodeId in LiteGraph.registered_node_types) {
-			const node = LiteGraph.registered_node_types[nodeId];
-			const nodeDef = defs[nodeId];
-			if(!nodeDef) continue;
-
-			node.nodeData = nodeDef;
+		for (const nodeId in defs) {
+			this.registerNodeDef(nodeId, defs[nodeId]);
 		}
 
 		for(let nodeNum in this.graph._nodes) {