diff --git a/__init__.py b/__init__.py index cadb7494..5fc4892c 100644 --- a/__init__.py +++ b/__init__.py @@ -18,7 +18,7 @@ import re import signal import nodes -version = "V1.10.2" +version = "V1.10.3" print(f"### Loading: ComfyUI-Manager ({version})") required_comfyui_revision = 1793 diff --git a/js/comfyui-manager.js b/js/comfyui-manager.js index 44358a8d..47dc10eb 100644 --- a/js/comfyui-manager.js +++ b/js/comfyui-manager.js @@ -279,8 +279,9 @@ function getNickname(node, nodename) { } } -function drawBadge(node, orig, ctx) { - const r = orig?.apply?.(node, arguments); +function drawBadge(node, orig, restArgs) { + let ctx = restArgs[0]; + const r = orig?.apply?.(node, restArgs); if (!node.flags.collapsed && badge_mode != 'none' && node.constructor.title_mode != LiteGraph.NO_TITLE) { let text = ""; @@ -953,7 +954,9 @@ app.registerExtension({ if(!node.badge_enabled) { node.getNickname = function () { return getNickname(node, node.comfyClass.trim()) }; const orig = node.__proto__.onDrawForeground; - node.onDrawForeground = function (ctx) { drawBadge(node, orig, ctx) }; + node.onDrawForeground = function (ctx) { + drawBadge(node, orig, arguments) + }; node.badge_enabled = true; } }, @@ -961,7 +964,7 @@ app.registerExtension({ if(!node.badge_enabled) { const orig = node.onDrawForeground; node.getNickname = function () { return getNickname(node, node.type.trim()) }; - node.onDrawForeground = function (ctx) { drawBadge(node, orig, ctx) }; + node.onDrawForeground = function (ctx) { drawBadge(node, orig, arguments) }; } },