uploadImage.js 328 Bytes
Newer Older
pythongosssss's avatar
pythongosssss committed
1
2
import { app } from "/scripts/app.js";

pythongosssss's avatar
pythongosssss committed
3
4
// Adds an upload button to the nodes

pythongosssss's avatar
pythongosssss committed
5
6
7
app.registerExtension({
	name: "Comfy.UploadImage",
	async beforeRegisterNodeDef(nodeType, nodeData, app) {
pythongosssss's avatar
pythongosssss committed
8
		if (nodeData.name === "LoadImage" || nodeData.name === "LoadImageMask") {
pythongosssss's avatar
pythongosssss committed
9
10
11
12
			nodeData.input.required.upload = ["IMAGEUPLOAD"];
		}
	},
});