|
|
@ -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"]
|
|
|
|