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): use refiner's seamless node for i2l VAE

Browse Source 
Closes  #5995
This commit is contained in:
psychedelicious 2024-03-20 15:41:49 +11:00
parent bf852348aa
commit c1de129bbc
1 changed files with 45 additions and 6 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

51
invokeai/frontend/web/src/features/nodes/util/graph/addVAEToGraph.ts
View File

@ -18,6 +18,7 @@ import {
SDXL_CANVAS_TEXT_TO_IMAGE_GRAPH, SDXL_CANVAS_TEXT_TO_IMAGE_GRAPH,
SDXL_IMAGE_TO_IMAGE_GRAPH, SDXL_IMAGE_TO_IMAGE_GRAPH,
SDXL_REFINER_INPAINT_CREATE_MASK, SDXL_REFINER_INPAINT_CREATE_MASK,
SDXL_REFINER_SEAMLESS,
SDXL_TEXT_TO_IMAGE_GRAPH, SDXL_TEXT_TO_IMAGE_GRAPH,
SEAMLESS, SEAMLESS,
TEXT_TO_IMAGE_GRAPH, TEXT_TO_IMAGE_GRAPH,
@ -38,6 +39,8 @@ export const addVAEToGraph = async (
const isAutoVae = !vae; const isAutoVae = !vae;
const isSeamlessEnabled = seamlessXAxis || seamlessYAxis; const isSeamlessEnabled = seamlessXAxis || seamlessYAxis;
const isSDXL = Boolean(graph.id?.includes('sdxl'));
const isUsingRefiner = isSDXL && Boolean(refinerModel);
if (!isAutoVae && !isSeamlessEnabled) { if (!isAutoVae && !isSeamlessEnabled) {
graph.nodes[VAE_LOADER] = { graph.nodes[VAE_LOADER] = {
@ -56,7 +59,13 @@ export const addVAEToGraph = async (
) { ) {
graph.edges.push({ graph.edges.push({
source: { source: {
node_id: isSeamlessEnabled ? SEAMLESS : isAutoVae ? modelLoaderNodeId : VAE_LOADER, node_id: isSeamlessEnabled
? isUsingRefiner
? SDXL_REFINER_SEAMLESS
: SEAMLESS
: isAutoVae
? modelLoaderNodeId
: VAE_LOADER,
field: 'vae', field: 'vae',
}, },
destination: { destination: {
@ -74,7 +83,13 @@ export const addVAEToGraph = async (
) { ) {
graph.edges.push({ graph.edges.push({
source: { source: {
node_id: isSeamlessEnabled ? SEAMLESS : isAutoVae ? modelLoaderNodeId : VAE_LOADER, node_id: isSeamlessEnabled
? isUsingRefiner
? SDXL_REFINER_SEAMLESS
: SEAMLESS
: isAutoVae
? modelLoaderNodeId
: VAE_LOADER,
field: 'vae', field: 'vae',
}, },
destination: { destination: {
@ -92,7 +107,13 @@ export const addVAEToGraph = async (
) { ) {
graph.edges.push({ graph.edges.push({
source: { source: {
node_id: isSeamlessEnabled ? SEAMLESS : isAutoVae ? modelLoaderNodeId : VAE_LOADER, node_id: isSeamlessEnabled
? isUsingRefiner
? SDXL_REFINER_SEAMLESS
: SEAMLESS
: isAutoVae
? modelLoaderNodeId
: VAE_LOADER,
field: 'vae', field: 'vae',
}, },
destination: { destination: {
@ -111,7 +132,13 @@ export const addVAEToGraph = async (
graph.edges.push( graph.edges.push(
{ {
source: { source: {
node_id: isSeamlessEnabled ? SEAMLESS : isAutoVae ? modelLoaderNodeId : VAE_LOADER, node_id: isSeamlessEnabled
? isUsingRefiner
? SDXL_REFINER_SEAMLESS
: SEAMLESS
: isAutoVae
? modelLoaderNodeId
: VAE_LOADER,
field: 'vae', field: 'vae',
}, },
destination: { destination: {
@ -122,7 +149,13 @@ export const addVAEToGraph = async (
{ {
source: { source: {
node_id: isSeamlessEnabled ? SEAMLESS : isAutoVae ? modelLoaderNodeId : VAE_LOADER, node_id: isSeamlessEnabled
? isUsingRefiner
? SDXL_REFINER_SEAMLESS
: SEAMLESS
: isAutoVae
? modelLoaderNodeId
: VAE_LOADER,
field: 'vae', field: 'vae',
}, },
destination: { destination: {
@ -137,7 +170,13 @@ export const addVAEToGraph = async (
if (graph.id === SDXL_CANVAS_INPAINT_GRAPH || graph.id === SDXL_CANVAS_OUTPAINT_GRAPH) { if (graph.id === SDXL_CANVAS_INPAINT_GRAPH || graph.id === SDXL_CANVAS_OUTPAINT_GRAPH) {
graph.edges.push({ graph.edges.push({
source: { source: {
node_id: isSeamlessEnabled ? SEAMLESS : isAutoVae ? modelLoaderNodeId : VAE_LOADER, node_id: isSeamlessEnabled
? isUsingRefiner
? SDXL_REFINER_SEAMLESS
: SEAMLESS
: isAutoVae
? modelLoaderNodeId
: VAE_LOADER,
field: 'vae', field: 'vae',
}, },
destination: { destination: {