Skip to content

Commit

Permalink
fix processor error
Browse files Browse the repository at this point in the history
  • Loading branch information
Vovanm88 committed Aug 21, 2024
1 parent 88bb0d0 commit 92db594
Showing 1 changed file with 8 additions and 7 deletions.
15 changes: 8 additions & 7 deletions sampling.py
Original file line number Diff line number Diff line change
Expand Up @@ -39,13 +39,14 @@ def model_forward(
if block_controlnet_hidden_states is not None:
controlnet_depth = len(block_controlnet_hidden_states)
for index_block, block in enumerate(model.double_blocks):
if isinstance(block.processor, DoubleStreamMixerProcessor):
if neg_mode:
for ip in block.processor.ip_adapters:
ip.ip_hidden_states = ip.in_hidden_states_neg
else:
for ip in block.processor.ip_adapters:
ip.ip_hidden_states = ip.in_hidden_states_pos
if hasattr(block, "processor"):
if isinstance(block.processor, DoubleStreamMixerProcessor):
if neg_mode:
for ip in block.processor.ip_adapters:
ip.ip_hidden_states = ip.in_hidden_states_neg
else:
for ip in block.processor.ip_adapters:
ip.ip_hidden_states = ip.in_hidden_states_pos

img, txt = block(img=img, txt=txt, vec=vec, pe=pe)
# controlnet residual
Expand Down

0 comments on commit 92db594

Please sign in to comment.