This commit is contained in:
Mary Hipp 2024-07-19 16:27:40 -04:00 committed by psychedelicious
parent 7854d913b2
commit 5060bf2f62
2 changed files with 6 additions and 7 deletions

View File

@ -1,6 +1,8 @@
import type { RootState } from 'app/store/store'; import type { RootState } from 'app/store/store';
import { fetchModelConfigWithTypeGuard } from 'features/metadata/util/modelFetchingHelpers';
import type { GraphType } from 'features/nodes/util/graph/generation/Graph'; import type { GraphType } from 'features/nodes/util/graph/generation/Graph';
import { Graph } from 'features/nodes/util/graph/generation/Graph'; import { Graph } from 'features/nodes/util/graph/generation/Graph';
import { isNonRefinerMainModelConfig, isSpandrelImageToImageModelConfig } from 'services/api/types';
import { assert } from 'tsafe'; import { assert } from 'tsafe';
import { import {
@ -22,8 +24,6 @@ import {
import { addLoRAs } from './generation/addLoRAs'; 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';
import { fetchModelConfigWithTypeGuard } from '../../../metadata/util/modelFetchingHelpers';
import { isNonRefinerMainModelConfig, isSpandrelImageToImageModelConfig } from '../../../../services/api/types';
export const UPSCALE_SCALE = 2; export const UPSCALE_SCALE = 2;
@ -38,8 +38,8 @@ export const buildMultidiffusionUpscsaleGraph = async (state: RootState): Promis
assert(upscaleInitialImage, 'No initial image found in state'); assert(upscaleInitialImage, 'No initial image found in state');
assert(tileControlnetModel, 'Tile controlnet is required'); assert(tileControlnetModel, 'Tile controlnet is required');
const outputWidth = ((upscaleInitialImage.width * UPSCALE_SCALE) / 8) * 8 const outputWidth = ((upscaleInitialImage.width * UPSCALE_SCALE) / 8) * 8;
const outputHeight = ((upscaleInitialImage.height * UPSCALE_SCALE) / 8) * 8 const outputHeight = ((upscaleInitialImage.height * UPSCALE_SCALE) / 8) * 8;
const g = new Graph(); const g = new Graph();
@ -69,7 +69,6 @@ export const buildMultidiffusionUpscsaleGraph = async (state: RootState): Promis
g.addEdge(upscaleNode, 'image', unsharpMaskNode2, 'image'); g.addEdge(upscaleNode, 'image', unsharpMaskNode2, 'image');
const resizeNode = g.addNode({ const resizeNode = g.addNode({
id: RESIZE, id: RESIZE,
type: 'img_resize', type: 'img_resize',
@ -211,7 +210,7 @@ export const buildMultidiffusionUpscsaleGraph = async (state: RootState): Promis
upscale_model: Graph.getModelMetadataField(upscaleModelConfig), upscale_model: Graph.getModelMetadataField(upscaleModelConfig),
creativity, creativity,
sharpness, sharpness,
structure structure,
}); });
} }

View File

@ -1,8 +1,8 @@
import { Flex, Text } from '@invoke-ai/ui-library'; import { Flex, Text } from '@invoke-ai/ui-library';
import { useAppSelector } from 'app/store/storeHooks'; import { useAppSelector } from 'app/store/storeHooks';
import { UPSCALE_SCALE } from 'features/nodes/util/graph/buildMultidiffusionUpscaleGraph';
import { useMemo } from 'react'; import { useMemo } from 'react';
import { useTranslation } from 'react-i18next'; import { useTranslation } from 'react-i18next';
import { UPSCALE_SCALE } from '../../../nodes/util/graph/buildMultidiffusionUpscaleGraph';
export const UpscaleSizeDetails = () => { export const UpscaleSizeDetails = () => {
const { t } = useTranslation(); const { t } = useTranslation();