mirror of
https://github.com/comfyanonymous/ComfyUI.git
synced 2026-01-15 00:30:55 +08:00
19 lines
426 B
JavaScript
19 lines
426 B
JavaScript
import { app } from "/scripts/app.js";
|
|
|
|
// Adds an upload button to the nodes
|
|
|
|
app.registerExtension({
|
|
name: "Comfy.UploadImage",
|
|
async beforeRegisterNodeDef(nodeType, nodeData, app) {
|
|
switch (nodeData.name) {
|
|
case "LoadImage":
|
|
case "LoadImageMask":
|
|
nodeData.input.required.upload = ["IMAGEUPLOAD"];
|
|
break;
|
|
case "LoadImageBatch":
|
|
nodeData.input.required.upload = ["MULTIIMAGEUPLOAD"];
|
|
break;
|
|
}
|
|
},
|
|
});
|