fix imageupload widget requiring a specific name

This commit is contained in:
pythongosssss 2023-11-15 20:22:19 +00:00
parent 2e79919b5a
commit 31bc856aa3
2 changed files with 4 additions and 1 deletions

View File

@ -189,6 +189,9 @@ function buildNodeDef(config, nodeName, defs, source = "workflow") {
if (inputName === "seed" || inputName === "noise_seed") {
inputDef = [...inputDef];
inputDef[1] = { control_after_generate: `${prefix} control_after_generate`, ...inputDef[1] };
} else if (inputDef[0] === "IMAGEUPLOAD") {
inputDef = [...inputDef];
inputDef[1] = { widget: `${prefix} ${inputDef[1]?.widget ?? "image"}`, ...inputDef[1] };
}
newDef.input.required[name] = inputDef;
}

View File

@ -376,7 +376,7 @@ export const ComfyWidgets = {
return { widget: node.addWidget("combo", inputName, defaultValue, () => {}, { values: type }) };
},
IMAGEUPLOAD(node, inputName, inputData, app) {
const imageWidget = node.widgets.find((w) => w.name === "image");
const imageWidget = node.widgets.find((w) => w.name === (inputData[1]?.widget ?? "image"));
let uploadWidget;
function showImage(name) {