From 82d3652d4afd5db472076657737faeb443c96fa9 Mon Sep 17 00:00:00 2001
From: Extraltodeus <extraltodeus@gmail.com>
Date: Wed, 4 Sep 2024 03:45:38 +0200
Subject: [PATCH] Update skimmed_CFG.py

---
 skimmed_CFG.py | 20 ++++++++++++++++++++
 1 file changed, 20 insertions(+)

diff --git a/skimmed_CFG.py b/skimmed_CFG.py
index 0e0f1a1..107c5a8 100644
--- a/skimmed_CFG.py
+++ b/skimmed_CFG.py
@@ -44,6 +44,8 @@ class CFG_skimming_single_scale_pre_cfg_node:
         start_at_sigma = model_sampling.percent_to_sigma(start_at_percentage)
         end_at_sigma   = model_sampling.percent_to_sigma(end_at_percentage)
         flip_at_sigma  = model_sampling.percent_to_sigma(flip_at_percentage)
+        if 1 > flip_at_percentage > 0:
+            print(f" \033[92mFlip at sigma: {round(flip_at_sigma, 2)}\033[0m")
 
         @torch.no_grad()
         def pre_cfg_patch(args):
@@ -86,6 +88,24 @@ class skimFlipPreCFGNode:
         m, = ssspcn.patch(model=model,Skimming_CFG=-1,full_skim_negative=True,disable_flipping_filter=reverse,flip_at_percentage=flip_at)
         return (m, )
 
+class constantSkimPreCFGNode:
+    @classmethod
+    def INPUT_TYPES(s):
+        return {"required": {
+                                "model": ("MODEL",),
+                                "enabled" : ("BOOLEAN", {"default": True}),
+                              }
+                              }
+    RETURN_TYPES = ("MODEL",)
+    FUNCTION = "patch"
+    CATEGORY = "model_patches/Pre CFG"
+    def patch(self, model, enabled):
+        if not enabled:
+            return model,
+        ssspcn = CFG_skimming_single_scale_pre_cfg_node()
+        m, = ssspcn.patch(model=model,Skimming_CFG=-1,full_skim_negative=True,disable_flipping_filter=False)
+        return (m, )
+
 class skimReplacePreCFGNode:
     @classmethod
     def INPUT_TYPES(s):