From c9c150f85009d8db7f6f1614763acd32b17ca3c7 Mon Sep 17 00:00:00 2001
From: psychedelicious <4822129+psychedelicious@users.noreply.github.com>
Date: Fri, 9 Feb 2024 18:51:26 +1100
Subject: [PATCH] feat(ui): use cfgRescaleMultiplier on canvas graphs

---
 .../src/features/nodes/util/graph/buildCanvasInpaintGraph.ts   | 3 +++
 .../src/features/nodes/util/graph/buildCanvasOutpaintGraph.ts  | 3 +++
 .../features/nodes/util/graph/buildCanvasSDXLInpaintGraph.ts   | 3 +++
 .../features/nodes/util/graph/buildCanvasSDXLOutpaintGraph.ts  | 3 +++
 4 files changed, 12 insertions(+)

diff --git a/invokeai/frontend/web/src/features/nodes/util/graph/buildCanvasInpaintGraph.ts b/invokeai/frontend/web/src/features/nodes/util/graph/buildCanvasInpaintGraph.ts
index 407581c181..bb52a44a8e 100644
--- a/invokeai/frontend/web/src/features/nodes/util/graph/buildCanvasInpaintGraph.ts
+++ b/invokeai/frontend/web/src/features/nodes/util/graph/buildCanvasInpaintGraph.ts
@@ -58,6 +58,7 @@ export const buildCanvasInpaintGraph = (
     negativePrompt,
     model,
     cfgScale: cfg_scale,
+    cfgRescaleMultiplier: cfg_rescale_multiplier,
     scheduler,
     steps,
     img2imgStrength: strength,
@@ -152,6 +153,7 @@ export const buildCanvasInpaintGraph = (
         is_intermediate,
         steps: steps,
         cfg_scale: cfg_scale,
+        cfg_rescale_multiplier,
         scheduler: scheduler,
         denoising_start: 1 - strength,
         denoising_end: 1,
@@ -175,6 +177,7 @@ export const buildCanvasInpaintGraph = (
         is_intermediate,
         steps: canvasCoherenceSteps,
         cfg_scale: cfg_scale,
+        cfg_rescale_multiplier,
         scheduler: scheduler,
         denoising_start: 1 - canvasCoherenceStrength,
         denoising_end: 1,
diff --git a/invokeai/frontend/web/src/features/nodes/util/graph/buildCanvasOutpaintGraph.ts b/invokeai/frontend/web/src/features/nodes/util/graph/buildCanvasOutpaintGraph.ts
index 5614aa0604..b82b55cfee 100644
--- a/invokeai/frontend/web/src/features/nodes/util/graph/buildCanvasOutpaintGraph.ts
+++ b/invokeai/frontend/web/src/features/nodes/util/graph/buildCanvasOutpaintGraph.ts
@@ -60,6 +60,7 @@ export const buildCanvasOutpaintGraph = (
     negativePrompt,
     model,
     cfgScale: cfg_scale,
+    cfgRescaleMultiplier: cfg_rescale_multiplier,
     scheduler,
     steps,
     img2imgStrength: strength,
@@ -161,6 +162,7 @@ export const buildCanvasOutpaintGraph = (
         is_intermediate,
         steps: steps,
         cfg_scale: cfg_scale,
+        cfg_rescale_multiplier,
         scheduler: scheduler,
         denoising_start: 1 - strength,
         denoising_end: 1,
@@ -184,6 +186,7 @@ export const buildCanvasOutpaintGraph = (
         is_intermediate,
         steps: canvasCoherenceSteps,
         cfg_scale: cfg_scale,
+        cfg_rescale_multiplier,
         scheduler: scheduler,
         denoising_start: 1 - canvasCoherenceStrength,
         denoising_end: 1,
diff --git a/invokeai/frontend/web/src/features/nodes/util/graph/buildCanvasSDXLInpaintGraph.ts b/invokeai/frontend/web/src/features/nodes/util/graph/buildCanvasSDXLInpaintGraph.ts
index afc0fd9386..00fea9a37e 100644
--- a/invokeai/frontend/web/src/features/nodes/util/graph/buildCanvasSDXLInpaintGraph.ts
+++ b/invokeai/frontend/web/src/features/nodes/util/graph/buildCanvasSDXLInpaintGraph.ts
@@ -60,6 +60,7 @@ export const buildCanvasSDXLInpaintGraph = (
     negativePrompt,
     model,
     cfgScale: cfg_scale,
+    cfgRescaleMultiplier: cfg_rescale_multiplier,
     scheduler,
     steps,
     seed,
@@ -151,6 +152,7 @@ export const buildCanvasSDXLInpaintGraph = (
         is_intermediate,
         steps: steps,
         cfg_scale: cfg_scale,
+        cfg_rescale_multiplier,
         scheduler: scheduler,
         denoising_start: refinerModel ? Math.min(refinerStart, 1 - strength) : 1 - strength,
         denoising_end: refinerModel ? refinerStart : 1,
@@ -174,6 +176,7 @@ export const buildCanvasSDXLInpaintGraph = (
         is_intermediate,
         steps: canvasCoherenceSteps,
         cfg_scale: cfg_scale,
+        cfg_rescale_multiplier,
         scheduler: scheduler,
         denoising_start: 1 - canvasCoherenceStrength,
         denoising_end: 1,
diff --git a/invokeai/frontend/web/src/features/nodes/util/graph/buildCanvasSDXLOutpaintGraph.ts b/invokeai/frontend/web/src/features/nodes/util/graph/buildCanvasSDXLOutpaintGraph.ts
index 2d70dc219b..f85760d8f2 100644
--- a/invokeai/frontend/web/src/features/nodes/util/graph/buildCanvasSDXLOutpaintGraph.ts
+++ b/invokeai/frontend/web/src/features/nodes/util/graph/buildCanvasSDXLOutpaintGraph.ts
@@ -62,6 +62,7 @@ export const buildCanvasSDXLOutpaintGraph = (
     negativePrompt,
     model,
     cfgScale: cfg_scale,
+    cfgRescaleMultiplier: cfg_rescale_multiplier,
     scheduler,
     steps,
     seed,
@@ -160,6 +161,7 @@ export const buildCanvasSDXLOutpaintGraph = (
         is_intermediate,
         steps: steps,
         cfg_scale: cfg_scale,
+        cfg_rescale_multiplier,
         scheduler: scheduler,
         denoising_start: refinerModel ? Math.min(refinerStart, 1 - strength) : 1 - strength,
         denoising_end: refinerModel ? refinerStart : 1,
@@ -183,6 +185,7 @@ export const buildCanvasSDXLOutpaintGraph = (
         is_intermediate,
         steps: canvasCoherenceSteps,
         cfg_scale: cfg_scale,
+        cfg_rescale_multiplier,
         scheduler: scheduler,
         denoising_start: 1 - canvasCoherenceStrength,
         denoising_end: 1,