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): sdxl graph builder

Browse Source
This commit is contained in:
psychedelicious 2024-08-23 18:13:35 +10:00
parent 01b9ca78e4
commit fadd20fb8e
1 changed files with 9 additions and 6 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

15
invokeai/frontend/web/src/features/nodes/util/graph/generation/buildSDXLGraph.ts
View File

@ -207,7 +207,7 @@ export const buildSDXLGraph = async (
); );
} }
const controlNetCollector = g.createNode({ const controlNetCollector = g.addNode({
type: 'collect', type: 'collect',
id: getPrefixedId('control_net_collector'), id: getPrefixedId('control_net_collector'),
}); });
@ -220,11 +220,12 @@ export const buildSDXLGraph = async (
modelConfig.base modelConfig.base
); );
if (controlNetResult.addedControlNets > 0) { if (controlNetResult.addedControlNets > 0) {
g.addNode(controlNetCollector);
g.addEdge(controlNetCollector, 'collection', denoise, 'control'); g.addEdge(controlNetCollector, 'collection', denoise, 'control');
} else {
g.deleteNode(controlNetCollector.id);
} }
const t2iAdapterCollector = g.createNode({ const t2iAdapterCollector = g.addNode({
type: 'collect', type: 'collect',
id: getPrefixedId('t2i_adapter_collector'), id: getPrefixedId('t2i_adapter_collector'),
}); });
@ -237,11 +238,12 @@ export const buildSDXLGraph = async (
modelConfig.base modelConfig.base
); );
if (t2iAdapterResult.addedT2IAdapters > 0) { if (t2iAdapterResult.addedT2IAdapters > 0) {
g.addNode(t2iAdapterCollector);
g.addEdge(t2iAdapterCollector, 'collection', denoise, 't2i_adapter'); g.addEdge(t2iAdapterCollector, 'collection', denoise, 't2i_adapter');
} else {
g.deleteNode(t2iAdapterCollector.id);
} }
const ipAdapterCollector = g.createNode({ const ipAdapterCollector = g.addNode({
type: 'collect', type: 'collect',
id: getPrefixedId('ip_adapter_collector'), id: getPrefixedId('ip_adapter_collector'),
}); });
@ -264,8 +266,9 @@ export const buildSDXLGraph = async (
const totalIPAdaptersAdded = const totalIPAdaptersAdded =
ipAdapterResult.addedIPAdapters + regionsResult.reduce((acc, r) => acc + r.addedIPAdapters, 0); ipAdapterResult.addedIPAdapters + regionsResult.reduce((acc, r) => acc + r.addedIPAdapters, 0);
if (totalIPAdaptersAdded > 0) { if (totalIPAdaptersAdded > 0) {
g.addNode(ipAdapterCollector);
g.addEdge(ipAdapterCollector, 'collection', denoise, 'ip_adapter'); g.addEdge(ipAdapterCollector, 'collection', denoise, 'ip_adapter');
} else {
g.deleteNode(ipAdapterCollector.id);
} }
if (state.system.shouldUseNSFWChecker) { if (state.system.shouldUseNSFWChecker) {