From aa4b2005641714a9b0bb764f7aa34d55e65e2387 Mon Sep 17 00:00:00 2001
From: jakechai <61836864+jakechai@users.noreply.github.com>
Date: Fri, 4 Oct 2024 10:11:50 +0800
Subject: [PATCH] add SearchAndRecolor ControlStyle FastUpscale

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

diff --git a/__init__.py b/__init__.py
index 525cb18..b7b07e0 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
+from .stability_api import StabilityConservativeUpscale, StabilityCreativeUpscale, StabilityRemoveBackground, StabilityInpainting, StabilityErase, StabilityCore, StabilityImageUltra, StabilitySearchAndReplace, StabilityOutpainting, StabilitySD3, StabilityControlSketch, StabilityControlStructure, StabilitySearchAndRecolor, StabilityControlStyle, StabilityFastUpscale
 
 NODE_CLASS_MAPPINGS = {
     "Stability Conservative Upscale": StabilityConservativeUpscale,
@@ -13,4 +13,7 @@ NODE_CLASS_MAPPINGS = {
     "Stability SD3": StabilitySD3,
     "Stability Control Skech": StabilityControlSketch,
     "Stability Control Structure": StabilityControlStructure,
+    "Stability Search And Recolor": StabilitySearchAndRecolor,
+    "Stability Control Style": StabilityControlStyle,
+    "Stability Fast Upscale": StabilityFastUpscale,
 }