Files
ComfyUI_frontend/src/extensions/core/uploadImage.ts
2024-12-30 17:26:37 -05:00

15 lines
382 B
TypeScript

import { ComfyNodeDef } from '@/types/apiTypes'
import { app } from '../../scripts/app'
// Adds an upload button to the nodes
app.registerExtension({
name: 'Comfy.UploadImage',
beforeRegisterNodeDef(nodeType, nodeData: ComfyNodeDef) {
if (nodeData?.input?.required?.image?.[1]?.image_upload === true) {
nodeData.input.required.upload = ['IMAGEUPLOAD']
}
}
})