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): img2img

Browse Source
This commit is contained in:
psychedelicious 2024-08-07 23:00:30 +10:00
parent 49e48c3eb7
commit cd4e464bde
1 changed files with 4 additions and 5 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

9
invokeai/frontend/web/src/features/nodes/util/graph/generation/addImageToImage.ts
View File

@ -1,7 +1,7 @@
import type { CanvasManager } from 'features/controlLayers/konva/CanvasManager'; import type { CanvasManager } from 'features/controlLayers/konva/CanvasManager';
import type { CanvasV2State, Dimensions } from 'features/controlLayers/store/types'; import type { CanvasV2State, Dimensions } from 'features/controlLayers/store/types';
import type { Graph } from 'features/nodes/util/graph/generation/Graph'; import type { Graph } from 'features/nodes/util/graph/generation/Graph';
import { isEqual, pick } from 'lodash-es'; import { isEqual } from 'lodash-es';
import type { Invocation } from 'services/api/types'; import type { Invocation } from 'services/api/types';
export const addImageToImage = async ( export const addImageToImage = async (
@ -17,15 +17,14 @@ export const addImageToImage = async (
): Promise<Invocation<'img_resize' | 'l2i'>> => { ): Promise<Invocation<'img_resize' | 'l2i'>> => {
denoise.denoising_start = denoising_start; denoise.denoising_start = denoising_start;
const cropBbox = pick(bbox.rect, ['x', 'y', 'width', 'height']); const { image_name } = await manager.getCompositeLayerImageDTO(bbox.rect);
const initialImage = await manager.getInitialImage({ bbox: cropBbox });
if (!isEqual(scaledSize, originalSize)) { if (!isEqual(scaledSize, originalSize)) {
// Resize the initial image to the scaled size, denoise, then resize back to the original size // Resize the initial image to the scaled size, denoise, then resize back to the original size
const resizeImageToScaledSize = g.addNode({ const resizeImageToScaledSize = g.addNode({
id: 'initial_image_resize_in', id: 'initial_image_resize_in',
type: 'img_resize', type: 'img_resize',
image: { image_name: initialImage.image_name }, image: { image_name },
...scaledSize, ...scaledSize,
}); });
const i2l = g.addNode({ id: 'i2l', type: 'i2l' }); const i2l = g.addNode({ id: 'i2l', type: 'i2l' });
@ -44,7 +43,7 @@ export const addImageToImage = async (
return resizeImageToOriginalSize; return resizeImageToOriginalSize;
} else { } else {
// No need to resize, just denoise // No need to resize, just denoise
const i2l = g.addNode({ id: 'i2l', type: 'i2l', image: { image_name: initialImage.image_name } }); const i2l = g.addNode({ id: 'i2l', type: 'i2l', image: { image_name } });
g.addEdge(vaeSource, 'vae', i2l, 'vae'); g.addEdge(vaeSource, 'vae', i2l, 'vae');
g.addEdge(i2l, 'latents', denoise, 'latents'); g.addEdge(i2l, 'latents', denoise, 'latents');
return l2i; return l2i;