Update nodes.py

2024-03-26 19:39:15 +01:00
parent 7ad1d56db1
commit 8c405109a1
1 changed files with 4 additions and 3 deletions
@@ -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, )
+        m = advcfg.patch(model,color_balance,color_balance,"hard" if boost else "soft", boost, 6.86)[0]
+        return (m, )