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: LoRA's not working with new canvas refine

Browse Source
This commit is contained in:
blessedcoolant 2023-08-24 07:51:32 +12:00 committed by Kent Keirsey
parent 5eb6148336
commit f45ea1145d
2 changed files with 40 additions and 0 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

19
invokeai/frontend/web/src/features/nodes/util/graphBuilders/addLoRAsToGraph.ts
View File

@ -6,6 +6,9 @@ import {
MetadataAccumulatorInvocation, MetadataAccumulatorInvocation,
} from 'services/api/types'; } from 'services/api/types';
import { import {
CANVAS_INPAINT_GRAPH,
CANVAS_OUTPAINT_GRAPH,
CANVAS_REFINE_DENOISE_LATENTS,
CLIP_SKIP, CLIP_SKIP,
LORA_LOADER, LORA_LOADER,
MAIN_MODEL_LOADER, MAIN_MODEL_LOADER,
@ -136,6 +139,22 @@ export const addLoRAsToGraph = (
}, },
}); });
if (
graph.id &&
[CANVAS_INPAINT_GRAPH, CANVAS_OUTPAINT_GRAPH].includes(graph.id)
) {
graph.edges.push({
source: {
node_id: currentLoraNodeId,
field: 'unet',
},
destination: {
node_id: CANVAS_REFINE_DENOISE_LATENTS,
field: 'unet',
},
});
}
graph.edges.push({ graph.edges.push({
source: { source: {
node_id: currentLoraNodeId, node_id: currentLoraNodeId,

21
invokeai/frontend/web/src/features/nodes/util/graphBuilders/addSDXLLoRAstoGraph.ts
View File

@ -6,10 +6,13 @@ import {
SDXLLoraLoaderInvocation, SDXLLoraLoaderInvocation,
} from 'services/api/types'; } from 'services/api/types';
import { import {
CANVAS_REFINE_DENOISE_LATENTS,
LORA_LOADER, LORA_LOADER,
METADATA_ACCUMULATOR, METADATA_ACCUMULATOR,
NEGATIVE_CONDITIONING, NEGATIVE_CONDITIONING,
POSITIVE_CONDITIONING, POSITIVE_CONDITIONING,
SDXL_CANVAS_INPAINT_GRAPH,
SDXL_CANVAS_OUTPAINT_GRAPH,
SDXL_MODEL_LOADER, SDXL_MODEL_LOADER,
} from './constants'; } from './constants';
@ -163,6 +166,24 @@ export const addSDXLLoRAsToGraph = (
}, },
}); });
if (
graph.id &&
[SDXL_CANVAS_INPAINT_GRAPH, SDXL_CANVAS_OUTPAINT_GRAPH].includes(
graph.id
)
) {
graph.edges.push({
source: {
node_id: currentLoraNodeId,
field: 'unet',
},
destination: {
node_id: CANVAS_REFINE_DENOISE_LATENTS,
field: 'unet',
},
});
}
graph.edges.push({ graph.edges.push({
source: { source: {
node_id: currentLoraNodeId, node_id: currentLoraNodeId,