mirror of
https://github.com/AUTOMATIC1111/stable-diffusion-webui.git
synced 2024-06-07 21:20:49 +00:00
change skip_early_cond code to use float
This commit is contained in:
parent
83182d2799
commit
6e9b69a338
@ -212,7 +212,7 @@ class CFGDenoiser(torch.nn.Module):
|
|||||||
uncond = denoiser_params.text_uncond
|
uncond = denoiser_params.text_uncond
|
||||||
skip_uncond = False
|
skip_uncond = False
|
||||||
|
|
||||||
if self.step < shared.opts.skip_cond_steps:
|
if self.step / self.total_steps <= shared.opts.skip_early_cond:
|
||||||
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]
|
||||||
|
Loading…
Reference in New Issue
Block a user