Skip SAG when latent is too small.

jn-jairo · Jan 10, 2024 · b4e915e · b4e915e
1 parent 1a57423
commit b4e915e
Showing 1 changed file with 2 additions and 0 deletions.
diff --git a/comfy_extras/nodes_sag.py b/comfy_extras/nodes_sag.py
@@ -143,6 +143,8 @@ def post_cfg_function(args):
             sigma = args["sigma"]
             model_options = args["model_options"]
             x = args["input"]
+            if min(cfg_result.shape[2:]) <= 4: #skip when too small to add padding
+                return cfg_result
 
             # create the adversarially blurred image
             degraded = create_blur_map(uncond_pred, uncond_attn, sag_sigma, sag_threshold)