Commit 794dd206 authored by comfyanonymous's avatar comfyanonymous
Browse files

Fix typo.

parent 0a6fd49a
...@@ -633,7 +633,7 @@ class UNetModel(nn.Module): ...@@ -633,7 +633,7 @@ class UNetModel(nn.Module):
for id, module in enumerate(self.output_blocks): for id, module in enumerate(self.output_blocks):
transformer_options["block"] = ("output", id) transformer_options["block"] = ("output", id)
hsp = hs.pop() hsp = hs.pop()
h = apply_control(h, control, 'output') hsp = apply_control(hsp, control, 'output')
if "output_block_patch" in transformer_patches: if "output_block_patch" in transformer_patches:
patch = transformer_patches["output_block_patch"] patch = transformer_patches["output_block_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