Fix typo.

This commit is contained in:
comfyanonymous 2023-11-07 23:41:55 -05:00
parent 0a6fd49a3e
commit 794dd2064d

View File

@ -633,7 +633,7 @@ class UNetModel(nn.Module):
for id, module in enumerate(self.output_blocks):
transformer_options["block"] = ("output", id)
hsp = hs.pop()
h = apply_control(h, control, 'output')
hsp = apply_control(hsp, control, 'output')
if "output_block_patch" in transformer_patches:
patch = transformer_patches["output_block_patch"]