upstream-archive/stable-diffusion-webui-forge
1
0
Fork 0
mirror of https://github.com/lllyasviel/stable-diffusion-webui-forge.git synced 2026-07-23 11:20:52 -05:00
Code Issues Packages Projects Releases Wiki Activity

SAG - add check for hires CFG 1 (#2434)

Browse Source
This commit is contained in:
DenOfEquity
2024-12-08 12:30:26 +00:00
committed by GitHub
parent 85a7db3c0f
commit 2e867aaf3a
1 changed files with 6 additions and 3 deletions
Download Patch File Download Diff File Expand all files Collapse all files

9
extensions-builtin/sd_forge_sag/scripts/forge_sag.py
View File

@@ -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