From b64e6565fbe6403b041c91910d89689d07822c3c Mon Sep 17 00:00:00 2001 From: Chaoses-Ib Date: Thu, 31 Oct 2024 08:54:46 +0800 Subject: [PATCH] fix(runtime/virtual): extra bool inputs (#22, #80) --- src/comfy_script/runtime/data/__init__.py | 4 +++- 1 file changed, 3 insertions(+), 1 deletion(-) diff --git a/src/comfy_script/runtime/data/__init__.py b/src/comfy_script/runtime/data/__init__.py index 628c29c..ccff50d 100644 --- a/src/comfy_script/runtime/data/__init__.py +++ b/src/comfy_script/runtime/data/__init__.py @@ -72,7 +72,9 @@ def _update_prompt(self, prompt: dict, id: IdManager) -> str: if group is not None and k in group: input_type = group[k][0] break - if factory.is_bool_enum(input_type): + # input_type is None if the input is extra + # e.g. ComfyUI-VideoHelperSuite (#22) + if input_type is not None and factory.is_bool_enum(input_type): prompt_inputs[k] = factory.to_bool_enum(input_type, v) else: prompt_inputs[k] = v