Commit baaf39b6 authored by AUTOMATIC1111's avatar AUTOMATIC1111

fix the typo -- thanks Cyberbeing

parent 757dda9a
...@@ -94,7 +94,7 @@ class CFGDenoiser(torch.nn.Module): ...@@ -94,7 +94,7 @@ class CFGDenoiser(torch.nn.Module):
def pad_cond_uncond(self, cond, uncond): def pad_cond_uncond(self, cond, uncond):
empty = shared.sd_model.cond_stage_model_empty_prompt empty = shared.sd_model.cond_stage_model_empty_prompt
num_repeats = (cond.shape[1] - cond.shape[1]) // empty.shape[1] num_repeats = (cond.shape[1] - uncond.shape[1]) // empty.shape[1]
if num_repeats < 0: if num_repeats < 0:
cond = pad_cond(cond, -num_repeats, empty) cond = pad_cond(cond, -num_repeats, empty)
......
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