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) {
graph.edges.push({
source: {
node_id: isAutoVae ? modelLoaderNodeId : VAE_LOADER,
field: isAutoVae && isOnnxModel ? 'vae_decoder' : 'vae',
},
destination: {
node_id: SDXL_REFINER_INPAINT_CREATE_MASK,
field: 'vae',
},
});
if (
graph.id === SDXL_CANVAS_INPAINT_GRAPH ||
graph.id === SDXL_CANVAS_OUTPAINT_GRAPH
) {
graph.edges.push({
source: {
node_id: isAutoVae ? modelLoaderNodeId : VAE_LOADER,
field: isAutoVae && isOnnxModel ? 'vae_decoder' : 'vae',
},
destination: {
node_id: SDXL_REFINER_INPAINT_CREATE_MASK,
field: 'vae',
},
});
}
}
if (vae && metadataAccumulator) {