Mirrored_Repos/InvokeAI
1
0
Fork 0
mirror of https://github.com/invoke-ai/InvokeAI synced 2024-08-30 20:32:17 +00:00
Code Issues Packages Projects Releases Wiki Activity

fix(ui): fix batch metadata logic when graph has no metadata

Browse Source 
On canvas, images have no metadata yet, so this needs to be handled
This commit is contained in:
psychedelicious 2023-10-18 22:33:17 +11:00
parent 52fbd1b222
commit 301a8fef92
2 changed files with 50 additions and 32 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

12
invokeai/frontend/web/src/features/nodes/util/graphBuilders/buildLinearBatchConfig.ts
View File

@ -11,7 +11,7 @@ import {
NOISE,
POSITIVE_CONDITIONING,
} from './constants';
import { removeMetadata } from './metadata';
import { getHasMetadata, removeMetadata } from './metadata';
export const prepareLinearUIBatch = (
state: RootState,
@ -40,6 +40,7 @@ export const prepareLinearUIBatch = (
});
}
if (getHasMetadata(graph)) {
// add to metadata
removeMetadata(graph, 'seed');
zipped.push({
@ -47,6 +48,7 @@ export const prepareLinearUIBatch = (
field_name: 'seed',
items: seeds,
});
}
if (graph.nodes[CANVAS_COHERENCE_NOISE]) {
zipped.push({
@ -78,12 +80,14 @@ export const prepareLinearUIBatch = (
}
// add to metadata
if (getHasMetadata(graph)) {
removeMetadata(graph, 'seed');
firstBatchDatumList.push({
node_path: METADATA,
field_name: 'seed',
items: seeds,
});
}
if (graph.nodes[CANVAS_COHERENCE_NOISE]) {
firstBatchDatumList.push({
@ -108,12 +112,14 @@ export const prepareLinearUIBatch = (
}
// add to metadata
if (getHasMetadata(graph)) {
removeMetadata(graph, 'seed');
secondBatchDatumList.push({
node_path: METADATA,
field_name: 'seed',
items: seeds,
});
}
if (graph.nodes[CANVAS_COHERENCE_NOISE]) {
secondBatchDatumList.push({
@ -140,12 +146,14 @@ export const prepareLinearUIBatch = (
}
// add to metadata
if (getHasMetadata(graph)) {
removeMetadata(graph, 'positive_prompt');
firstBatchDatumList.push({
node_path: METADATA,
field_name: 'positive_prompt',
items: extendedPrompts,
});
}
if (shouldConcatSDXLStylePrompt && model?.base_model === 'sdxl') {
const stylePrompts = extendedPrompts.map((p) =>
@ -161,6 +169,7 @@ export const prepareLinearUIBatch = (
}
// add to metadata
if (getHasMetadata(graph)) {
removeMetadata(graph, 'positive_style_prompt');
firstBatchDatumList.push({
node_path: METADATA,
@ -168,6 +177,7 @@ export const prepareLinearUIBatch = (
items: extendedPrompts,
});
}
}
data.push(firstBatchDatumList);
}

8
invokeai/frontend/web/src/features/nodes/util/graphBuilders/metadata.ts
View File

@ -56,3 +56,11 @@ export const removeMetadata = (
delete metadataNode[key];
};
export const getHasMetadata = (graph: NonNullableGraph): boolean => {
const metadataNode = graph.nodes[METADATA] as
| CoreMetadataInvocation
| undefined;
return Boolean(metadataNode);
};