diff --git a/src/comfyui_ino_nodes/utils/sampler_helper.py b/src/comfyui_ino_nodes/utils/sampler_helper.py index eca6541..3c1a87d 100644 --- a/src/comfyui_ino_nodes/utils/sampler_helper.py +++ b/src/comfyui_ino_nodes/utils/sampler_helper.py @@ -12,6 +12,8 @@ from comfy_extras.nodes_flux import FluxGuidance from comfy.samplers import SAMPLER_NAMES, SCHEDULER_NAMES +from ..node_helper import any_typ + default_bool = ["unset", "true", "false"] sampler_names = ["unset"] + SAMPLER_NAMES scheduler_names = ["unset"] + SCHEDULER_NAMES @@ -495,6 +497,7 @@ class InoLoadSamplerModels: return { "required": { "enabled": ("BOOLEAN", {"default": True, "label_off": "OFF", "label_on": "ON"}), + "execute": (any_typ,), "model_config": ("STRING", { "multiline": True, }), @@ -525,7 +528,7 @@ class InoLoadSamplerModels: CATEGORY = "InoSamplerHelper" def function( - self,enabled, + self, enabled, execute, model_config, lora_1_config, lora_2_config, lora_3_config, lora_4_config, clip_device, @@ -534,6 +537,12 @@ class InoLoadSamplerModels: if not enabled: return (None, None, None, None, None, ) + if execute is None: + return (None, None, None, None, None, ) + + if not execute: + return (None, None, None, None, None, ) + load_json = InoJsonHelper.string_to_dict(model_config) if not load_json["success"]: return (None, None, None, None, None,) diff --git a/web/js/inonodes.js b/web/js/inonodes.js index ac21c18..199ede2 100644 --- a/web/js/inonodes.js +++ b/web/js/inonodes.js @@ -39,7 +39,5 @@ app.registerExtension({ }, async setup() { - // to keep Set/Get node virtual connections visible when offscreen - alert("Setup complete!") } });