Commit feeb6802 authored by drhead's avatar drhead Committed by GitHub

fix case where first step skilled if skip early cond is 0

parent 029adbe5
......@@ -212,7 +212,7 @@ class CFGDenoiser(torch.nn.Module):
uncond = denoiser_params.text_uncond
skip_uncond = False
if self.step / self.total_steps <= shared.opts.skip_early_cond:
if shared.opts.skip_early_cond != 0. and self.step / self.total_steps <= shared.opts.skip_early_cond:
skip_uncond = True
x_in = x_in[:-batch_size]
sigma_in = sigma_in[:-batch_size]
......
Markdown is supported
0% or
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment