fix(runtime/virtual): extra bool inputs (#22, #80)

Chaoses-Ib · Oct 31, 2024 · b64e656 · b64e656
1 parent bc654d3
commit b64e656
Showing 1 changed file with 3 additions and 1 deletion.
diff --git 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