diff --git a/invokeai/frontend/web/src/features/nodes/util/schema/parseSchema.ts b/invokeai/frontend/web/src/features/nodes/util/schema/parseSchema.ts index a09b341c286..32ceaffdf60 100644 --- a/invokeai/frontend/web/src/features/nodes/util/schema/parseSchema.ts +++ b/invokeai/frontend/web/src/features/nodes/util/schema/parseSchema.ts @@ -58,6 +58,9 @@ const isAllowedOutputField = (nodeType: string, fieldName: string) => { if (RESERVED_OUTPUT_FIELD_NAMES.includes(fieldName)) { return false; } + if (nodeType === 'image_batch' && fieldName !== 'image') { + return false; + } return true; };