mirror of
https://github.com/comfyanonymous/ComfyUI.git
synced 2025-03-15 14:09:36 +00:00
Skip SAG when latent is too small.
This commit is contained in:
parent
1a57423d30
commit
b4e915e745
@ -143,6 +143,8 @@ class SelfAttentionGuidance:
|
|||||||
sigma = args["sigma"]
|
sigma = args["sigma"]
|
||||||
model_options = args["model_options"]
|
model_options = args["model_options"]
|
||||||
x = args["input"]
|
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
|
# create the adversarially blurred image
|
||||||
degraded = create_blur_map(uncond_pred, uncond_attn, sag_sigma, sag_threshold)
|
degraded = create_blur_map(uncond_pred, uncond_attn, sag_sigma, sag_threshold)
|
||||||
|
Loading…
Reference in New Issue
Block a user