github/InvokeAI
1
1
Fork 0
mirror of https://github.com/invoke-ai/InvokeAI.git synced 2026-04-23 03:00:31 -04:00
Code Issues Packages Projects Releases Wiki Activity

feat(ui): support multiple kontext ref images in studio

Browse Source
This commit is contained in:
psychedelicious
2025-07-30 11:53:00 +10:00
parent afb67efef9
commit 415616d83f
2 changed files with 24 additions and 17 deletions
Download Patch File Download Diff File Expand all files Collapse all files

35
invokeai/frontend/web/src/features/nodes/util/graph/generation/buildFLUXGraph.ts
View File

@@ -154,22 +154,27 @@ export const buildFLUXGraph = async (arg: GraphBuilderArg): Promise<GraphBuilder
.filter((entity) => isFluxKontextReferenceImageConfig(entity.config))
.filter((entity) => getGlobalReferenceImageWarnings(entity, model).length === 0);
// FLUX Kontext supports only a single conditioning image - we'll just take the first one.
// In the future, we can explore concatenating multiple conditioning images in image or latent space.
const firstValidFLUXKontextConfig = validFLUXKontextConfigs[0];
if (firstValidFLUXKontextConfig) {
const { image } = firstValidFLUXKontextConfig.config;
assert(image, 'getGlobalReferenceImageWarnings checks if the image is there, this should never raise');
const kontextConditioning = g.addNode({
type: 'flux_kontext',
id: getPrefixedId('flux_kontext'),
image,
if (validFLUXKontextConfigs.length > 0) {
const kontextCollector = g.addNode({
id: getPrefixedId('flux_kontext_collector'),
type: 'collect',
});
g.addEdge(kontextConditioning, 'kontext_cond', denoise, 'kontext_conditioning');
g.upsertMetadata({ ref_images: [firstValidFLUXKontextConfig] }, 'merge');
g.addEdge(kontextCollector, 'collection', denoise, 'kontext_conditioning');
for (const kontextConfig of validFLUXKontextConfigs) {
const { image } = kontextConfig.config;
assert(image, 'getGlobalReferenceImageWarnings checks if the image is there, this should never raise');
const kontextConditioning = g.addNode({
type: 'flux_kontext',
id: getPrefixedId(`flux_kontext_${kontextConfig.id}`),
image,
});
g.addEdge(kontextConditioning, 'kontext_cond', kontextCollector, 'item');
}
g.upsertMetadata({ ref_images: [validFLUXKontextConfigs] }, 'merge');
}
}

6
invokeai/frontend/web/src/features/queue/store/readiness.ts
View File

@@ -282,7 +282,8 @@ const getReasonsWhyCannotEnqueueGenerateTab = (arg: {
// Flux Kontext only supports 1x Reference Image at a time.
if (
(model?.base === 'flux-kontext' || (model?.base === 'flux' && model?.name?.toLowerCase().includes('kontext'))) &&
model?.base === 'flux-kontext' &&
// (model?.base === 'flux-kontext' || (model?.base === 'flux' && model?.name?.toLowerCase().includes('kontext'))) &&
referenceImageCount > 1
) {
reasons.push({ content: i18n.t('parameters.invoke.fluxKontextMultipleReferenceImages') });
@@ -640,7 +641,8 @@ const getReasonsWhyCannotEnqueueCanvasTab = (arg: {
// Flux Kontext only supports 1x Reference Image at a time.
if (
(model?.base === 'flux-kontext' || (model?.base === 'flux' && model?.name?.toLowerCase().includes('kontext'))) &&
model?.base === 'flux-kontext' &&
// (model?.base === 'flux-kontext' || (model?.base === 'flux' && model?.name?.toLowerCase().includes('kontext'))) &&
referenceImageCount > 1
) {
reasons.push({ content: i18n.t('parameters.invoke.fluxKontextMultipleReferenceImages') });