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: Change graph id vals to constants

Browse Source
This commit is contained in:
blessedcoolant 2023-07-01 21:26:41 +12:00 committed by psychedelicious
parent 511978979e
commit 089d95baeb
1 changed files with 6 additions and 6 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

12
invokeai/frontend/web/src/features/nodes/util/graphBuilders/addVAEToGraph.ts
View File

@ -2,10 +2,13 @@ import { RootState } from 'app/store/store';
import { NonNullableGraph } from 'features/nodes/types/types'; import { NonNullableGraph } from 'features/nodes/types/types';
import { modelIdToVAEModelField } from '../modelIdToVAEModelField'; import { modelIdToVAEModelField } from '../modelIdToVAEModelField';
import { import {
IMAGE_TO_IMAGE_GRAPH,
IMAGE_TO_LATENTS, IMAGE_TO_LATENTS,
INPAINT, INPAINT,
INPAINT_GRAPH,
LATENTS_TO_IMAGE, LATENTS_TO_IMAGE,
MAIN_MODEL_LOADER, MAIN_MODEL_LOADER,
TEXT_TO_IMAGE_GRAPH,
VAE_LOADER, VAE_LOADER,
} from './constants'; } from './constants';
@ -24,10 +27,7 @@ export const addVAEToGraph = (
}; };
} }
if ( if (graph.id === TEXT_TO_IMAGE_GRAPH || graph.id === IMAGE_TO_IMAGE_GRAPH) {
graph.id === 'text_to_image_graph' ||
graph.id === 'image_to_image_graph'
) {
graph.edges.push({ graph.edges.push({
source: { source: {
node_id: vaeId === 'auto' ? MAIN_MODEL_LOADER : VAE_LOADER, node_id: vaeId === 'auto' ? MAIN_MODEL_LOADER : VAE_LOADER,
@ -40,7 +40,7 @@ export const addVAEToGraph = (
}); });
} }
if (graph.id === 'image_to_image_graph') { if (graph.id === IMAGE_TO_IMAGE_GRAPH) {
graph.edges.push({ graph.edges.push({
source: { source: {
node_id: vaeId === 'auto' ? MAIN_MODEL_LOADER : VAE_LOADER, node_id: vaeId === 'auto' ? MAIN_MODEL_LOADER : VAE_LOADER,
@ -53,7 +53,7 @@ export const addVAEToGraph = (
}); });
} }
if (graph.id === 'inpaint_graph') { if (graph.id === INPAINT_GRAPH) {
graph.edges.push({ graph.edges.push({
source: { source: {
node_id: vaeId === 'auto' ? MAIN_MODEL_LOADER : VAE_LOADER, node_id: vaeId === 'auto' ? MAIN_MODEL_LOADER : VAE_LOADER,