From 2e867aaf3af58811d0d7203f8235e53a729237d3 Mon Sep 17 00:00:00 2001 From: DenOfEquity <166248528+DenOfEquity@users.noreply.github.com> Date: Sun, 8 Dec 2024 12:30:26 +0000 Subject: [PATCH] SAG - add check for hires CFG 1 (#2434) --- extensions-builtin/sd_forge_sag/scripts/forge_sag.py | 9 ++++++--- 1 file changed, 6 insertions(+), 3 deletions(-) diff --git a/extensions-builtin/sd_forge_sag/scripts/forge_sag.py b/extensions-builtin/sd_forge_sag/scripts/forge_sag.py index e0786d63..19eaceab 100644 --- a/extensions-builtin/sd_forge_sag/scripts/forge_sag.py +++ b/extensions-builtin/sd_forge_sag/scripts/forge_sag.py @@ -201,11 +201,14 @@ class SAGForForge(scripts.Script): # not for FLux if not shared.sd_model.is_webui_legacy_model(): # ideally would be is_flux - gr.Info ("Self Attention Guidance is not compatible with Flux") + print("Self Attention Guidance is not compatible with Flux") return # Self Attention Guidance errors if CFG is 1 - if p.cfg_scale == 1: - gr.Info ("Self Attention Guidance requires CFG > 1") + if p.is_hr_pass == False and p.cfg_scale <= 1: + print("Self Attention Guidance requires CFG > 1") + return + if p.is_hr_pass == True and p.hr_cfg <= 1: + print("Self Attention Guidance (hires pass) requires Hires CFG > 1") return unet = p.sd_model.forge_objects.unet