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

restore scale but hardcode it to 2 regardless of upscale model

Browse Source
This commit is contained in:
Mary Hipp 2024-07-19 11:52:11 -04:00 committed by psychedelicious
parent 5266e9e682
commit d166b08b6a
1 changed files with 18 additions and 5 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

23
invokeai/frontend/web/src/features/nodes/util/graph/buildMultidiffusionUpscaleGraph.ts
View File

@ -12,6 +12,7 @@ import {
NEGATIVE_CONDITIONING, NEGATIVE_CONDITIONING,
NOISE, NOISE,
POSITIVE_CONDITIONING, POSITIVE_CONDITIONING,
RESIZE,
SDXL_MODEL_LOADER, SDXL_MODEL_LOADER,
SPANDREL, SPANDREL,
TILED_MULTI_DIFFUSION_DENOISE_LATENTS, TILED_MULTI_DIFFUSION_DENOISE_LATENTS,
@ -22,6 +23,8 @@ import { addLoRAs } from './generation/addLoRAs';
import { addSDXLLoRas } from './generation/addSDXLLoRAs'; import { addSDXLLoRas } from './generation/addSDXLLoRAs';
import { getBoardField, getSDXLStylePrompts } from './graphBuilderUtils'; import { getBoardField, getSDXLStylePrompts } from './graphBuilderUtils';
const SCALE = 2;
export const buildMultidiffusionUpscsaleGraph = async (state: RootState): Promise<GraphType> => { export const buildMultidiffusionUpscsaleGraph = async (state: RootState): Promise<GraphType> => {
const { model, cfgScale: cfg_scale, scheduler, steps, vaePrecision, seed, vae } = state.generation; const { model, cfgScale: cfg_scale, scheduler, steps, vaePrecision, seed, vae } = state.generation;
const { positivePrompt, negativePrompt } = state.controlLayers.present; const { positivePrompt, negativePrompt } = state.controlLayers.present;
@ -61,14 +64,24 @@ export const buildMultidiffusionUpscsaleGraph = async (state: RootState): Promis
g.addEdge(upscaleNode, 'image', unsharpMaskNode2, 'image'); g.addEdge(upscaleNode, 'image', unsharpMaskNode2, 'image');
const resizeNode = g.addNode({
id: RESIZE,
type: 'img_resize',
width: ((upscaleInitialImage.width * SCALE) / 8) * 8,
height: ((upscaleInitialImage.height * SCALE) / 8) * 8,
resample_mode: 'lanczos',
});
g.addEdge(unsharpMaskNode2, 'image', resizeNode, 'image');
const noiseNode = g.addNode({ const noiseNode = g.addNode({
id: NOISE, id: NOISE,
type: 'noise', type: 'noise',
seed, seed,
}); });
g.addEdge(unsharpMaskNode2, 'width', noiseNode, 'width'); g.addEdge(resizeNode, 'width', noiseNode, 'width');
g.addEdge(unsharpMaskNode2, 'height', noiseNode, 'height'); g.addEdge(resizeNode, 'height', noiseNode, 'height');
const i2lNode = g.addNode({ const i2lNode = g.addNode({
id: IMAGE_TO_LATENTS, id: IMAGE_TO_LATENTS,
@ -77,7 +90,7 @@ export const buildMultidiffusionUpscsaleGraph = async (state: RootState): Promis
tiled: tiledVAE, tiled: tiledVAE,
}); });
g.addEdge(unsharpMaskNode2, 'image', i2lNode, 'image'); g.addEdge(resizeNode, 'image', i2lNode, 'image');
const l2iNode = g.addNode({ const l2iNode = g.addNode({
type: 'l2i', type: 'l2i',
@ -187,7 +200,7 @@ export const buildMultidiffusionUpscsaleGraph = async (state: RootState): Promis
end_step_percent: (structure + 10) * 0.025 + 0.3, end_step_percent: (structure + 10) * 0.025 + 0.3,
}); });
g.addEdge(unsharpMaskNode2, 'image', controlnetNode1, 'image'); g.addEdge(resizeNode, 'image', controlnetNode1, 'image');
const controlnetNode2 = g.addNode({ const controlnetNode2 = g.addNode({
id: 'controlnet_2', id: 'controlnet_2',
@ -200,7 +213,7 @@ export const buildMultidiffusionUpscsaleGraph = async (state: RootState): Promis
end_step_percent: 0.8, end_step_percent: 0.8,
}); });
g.addEdge(unsharpMaskNode2, 'image', controlnetNode2, 'image'); g.addEdge(resizeNode, 'image', controlnetNode2, 'image');
const collectNode = g.addNode({ const collectNode = g.addNode({
id: CONTROL_NET_COLLECT, id: CONTROL_NET_COLLECT,