Fix set workflow attribute in node

2023-05-20 23:22:56 -05:00
parent 3aad259869
commit 1014afc9ab
3 changed files with 15 additions and 7 deletions
--- a/src/lib/nodes/ComfyConfigureQueuePromptButton.ts
+++ b/src/lib/nodes/ComfyConfigureQueuePromptButton.ts
@@ -30,9 +30,9 @@ export default class ComfyConfigureQueuePromptButton extends ComfyGraphNode {
            }

            if (typeof param === "string")
-                this.workflow.attrs.queuePromptButtonName = param || ""
+                this.workflow.setAttribute("queuePromptButtonName", param || "")
            else if (typeof param === "object" && "buttonName" in param)
-                this.workflow.attrs.queuePromptButtonName = param.buttonName || ""
+                this.workflow.setAttribute("queuePromptButtonName", param.buttonName || "")
        }
    }
 }