Skip to content

GitLab

Commit 78d8035f authored by comfyanonymous's avatar comfyanonymous
Browse files

Fix bug with controlnet.

parent c9f5d5b2
Hide whitespace changes
Inline Side-by-side
Showing with 4 additions and 1 deletion
comfy/ldm/modules/attention.py
View file @ 78d8035f
...@@ -565,7 +565,10 @@ class BasicTransformerBlock(nn.Module): ...@@ -565,7 +565,10 @@ class BasicTransformerBlock(nn.Module):
for p in patch: for p in patch:
n, context_attn1, value_attn1 = p(n, context_attn1, value_attn1, extra_options) n, context_attn1, value_attn1 = p(n, context_attn1, value_attn1, extra_options)
transformer_block = (block[0], block[1], block_index) if block is not None:
transformer_block = (block[0], block[1], block_index)
else:
transformer_block = None
attn1_replace_patch = transformer_patches_replace.get("attn1", {}) attn1_replace_patch = transformer_patches_replace.get("attn1", {})
block_attn1 = transformer_block block_attn1 = transformer_block
if block_attn1 not in attn1_replace_patch: if block_attn1 not in attn1_replace_patch:
......
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