This commit is contained in:
Damian at mba
2022-10-18 13:52:40 +02:00
parent 056cb0d8a8
commit 711ffd238f
2 changed files with 43 additions and 28 deletions

View File

@ -44,7 +44,7 @@ class CFGDenoiser(nn.Module):
CrossAttentionControl.setup_attention_editing(self.inner_model, edited_conditioning, edit_opcodes)
else:
# pass through the attention func but don't act on it
CrossAttentionControl.setup_attention_editing(self.inner_model)
CrossAttentionControl.clear_attention_editing(self.inner_model)
def forward(self, x, sigma, uncond, cond, cond_scale):