implement option to skip uncond on all steps below ngms

This commit is contained in:
drhead
2024-04-23 03:15:56 -04:00
committed by GitHub
parent 33cbbf9f8b
commit 029adbe531

View File

@@ -218,7 +218,7 @@ class CFGDenoiser(torch.nn.Module):
sigma_in = sigma_in[:-batch_size] sigma_in = sigma_in[:-batch_size]
# alternating uncond allows for higher thresholds without the quality loss normally expected from raising it # alternating uncond allows for higher thresholds without the quality loss normally expected from raising it
if self.step % 2 and s_min_uncond > 0 and sigma[0] < s_min_uncond and not is_edit_model: if (self.step % 2 or shared.opts.s_min_uncond_all) and s_min_uncond > 0 and sigma[0] < s_min_uncond and not is_edit_model:
skip_uncond = True skip_uncond = True
x_in = x_in[:-batch_size] x_in = x_in[:-batch_size]
sigma_in = sigma_in[:-batch_size] sigma_in = sigma_in[:-batch_size]