From 8c405109a1c5ef72d22ad9fb85013d600c716a84 Mon Sep 17 00:00:00 2001
From: Extraltodeus <extraltodeus@gmail.com>
Date: Tue, 26 Mar 2024 19:39:15 +0100
Subject: [PATCH] Update nodes.py

---
 nodes.py | 7 ++++---
 1 file changed, 4 insertions(+), 3 deletions(-)

diff --git a/nodes.py b/nodes.py
index 889ec0f..031f8ee 100644
--- a/nodes.py
+++ b/nodes.py
@@ -149,13 +149,14 @@ class simpleDynamicCFG:
         return {"required": {
                                 "model": ("MODEL",),
                                 "boost" : ("BOOLEAN", {"default": True}),
+                                "color_balance" : ("BOOLEAN", {"default": True}),
                               }}
     RETURN_TYPES = ("MODEL",)
     FUNCTION = "patch"
 
     CATEGORY = "model_patches"
 
-    def patch(self, model, boost):
+    def patch(self, model, boost, color_balance):
         advcfg = advancedDynamicCFG()
-        m = advcfg.patch(model,False,False,"hard" if boost else "soft", boost, 6.86)[0]
-        return (m, )
\ No newline at end of file
+        m = advcfg.patch(model,color_balance,color_balance,"hard" if boost else "soft", boost, 6.86)[0]
+        return (m, )