Fix assertion when there is a node with a "type" parameter in a custo…

…m workflow #1473
Acly · Dec 10, 2024 · 4a044cf · 4a044cf
1 parent a6f0c9b
commit 4a044cf
Showing 1 changed file with 2 additions and 1 deletion.
diff --git a/ai_diffusion/custom_workflow.py b/ai_diffusion/custom_workflow.py
@@ -288,7 +288,8 @@ def compare(a: str, b: str):
 def workflow_parameters(w: ComfyWorkflow):
     text_types = ("text", "prompt (positive)", "prompt (negative)")
     for node in w:
-        match (node.type, node.input("type", "")):
+        param_type = node.input("type", "") if node.type == "ETN_Parameter" else ""
+        match (node.type, param_type):
             case ("ETN_KritaStyle", _):
                 name = node.input("name", "Style")
                 yield CustomParam(ParamKind.style, name, node.input("sampler_preset", "auto"))