From 50e2063dfe86241d396e288c9e7e06fd77fdd490 Mon Sep 17 00:00:00 2001
From: Jake Chai <61836864+jakechai@users.noreply.github.com>
Date: Thu, 27 Feb 2025 09:39:24 +0800
Subject: [PATCH] Add missing nodes.

---
 __init__.py | 9 +++++----
 1 file changed, 5 insertions(+), 4 deletions(-)

diff --git a/__init__.py b/__init__.py
index 221974d..5a3921a 100644
--- a/__init__.py
+++ b/__init__.py
@@ -1,4 +1,4 @@
-from .stability_api import StabilityConservativeUpscale, StabilityCreativeUpscale, StabilityRemoveBackground, StabilityInpainting, StabilityErase, StabilityCore, StabilityImageUltra, StabilitySearchAndReplace, StabilityOutpainting, StabilitySD3, StabilityControlSketch, StabilityControlStructure, StabilityFastUpscale, StabilityControlStyle, StabilitySearchAndRecolor
+from .stability_api import StabilityConservativeUpscale, StabilityCreativeUpscale, StabilityRemoveBackground, StabilityInpainting, StabilityErase, StabilityCore, StabilityImageUltra, StabilitySearchAndReplace, StabilityOutpainting, StabilitySD3, StabilityControlSketch, StabilityControlStructure, StabilitySearchAndRecolor, StabilityControlStyle, StabilityFastUpscale, StabilityReplaceRelight
 
 NODE_CLASS_MAPPINGS = {
     "Stability Conservative Upscale": StabilityConservativeUpscale,
@@ -13,7 +13,8 @@ NODE_CLASS_MAPPINGS = {
     "Stability SD3": StabilitySD3,
     "Stability Control Sketch": StabilityControlSketch,
     "Stability Control Structure": StabilityControlStructure,
+    "Stability Search And Recolor": StabilitySearchAndRecolor,
+    "Stability Control Style": StabilityControlStyle,
     "Stability Fast Upscale": StabilityFastUpscale,
-    "Stability Style": StabilityControlStyle,
-    "Stability Search and Recolor": StabilitySearchAndRecolor,
-}
\ No newline at end of file
+    "Stability Replace Background and Relight": StabilityReplaceRelight,
+}