fix: Create SDXL Refiner Create Mask only in inpaint & outpaint

This commit is contained in:
blessedcoolant 2023-08-31 06:33:09 +12:00
parent 97763f778a
commit 4bac36356a

View File

@ -149,16 +149,21 @@ export const addVAEToGraph = (
} }
if (shouldUseSDXLRefiner) { if (shouldUseSDXLRefiner) {
graph.edges.push({ if (
source: { graph.id === SDXL_CANVAS_INPAINT_GRAPH ||
node_id: isAutoVae ? modelLoaderNodeId : VAE_LOADER, graph.id === SDXL_CANVAS_OUTPAINT_GRAPH
field: isAutoVae && isOnnxModel ? 'vae_decoder' : 'vae', ) {
}, graph.edges.push({
destination: { source: {
node_id: SDXL_REFINER_INPAINT_CREATE_MASK, node_id: isAutoVae ? modelLoaderNodeId : VAE_LOADER,
field: 'vae', field: isAutoVae && isOnnxModel ? 'vae_decoder' : 'vae',
}, },
}); destination: {
node_id: SDXL_REFINER_INPAINT_CREATE_MASK,
field: 'vae',
},
});
}
} }
if (vae && metadataAccumulator) { if (vae && metadataAccumulator) {