298977887/ComfyUI
1
0
Fork 0
mirror of https://github.com/comfyanonymous/ComfyUI.git synced 2025-01-11 10:25:16 +00:00
Code Issues Packages Projects Releases Wiki Activity

Fix bug with controlnet.

Browse Source
This commit is contained in:
comfyanonymous 2023-06-24 11:02:38 -04:00
parent c9f5d5b2e1
commit 78d8035f73
1 changed files with 4 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

5
comfy/ldm/modules/attention.py
View File

@ -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: