Don't hardcode node names for image upload widget.

This commit is contained in:
comfyanonymous
2023-08-22 19:41:49 -04:00
parent afcb9cb1df
commit d7b3b0f8c1
2 changed files with 4 additions and 3 deletions

View File

@@ -5,7 +5,8 @@ import { app } from "../../scripts/app.js";
app.registerExtension({
name: "Comfy.UploadImage",
async beforeRegisterNodeDef(nodeType, nodeData, app) {
if (nodeData.name === "LoadImage" || nodeData.name === "LoadImageMask") {
console.log(nodeData);
if (nodeData?.input?.required?.image?.[1]?.image_upload === true) {
nodeData.input.required.upload = ["IMAGEUPLOAD"];
}
},