Fix redundant prev_output_channel assignment in UNet2DModel

ahmedbelgacem · web-flow · commit fc0eddb9fd79 · 2025-03-03T15:38:24.000+01:00
diff --git a/src/diffusers/models/unets/unet_2d.py b/src/diffusers/models/unets/unet_2d.py
@@ -240,7 +240,6 @@ def __init__(
                 dropout=dropout,
             )
             self.up_blocks.append(up_block)
-            prev_output_channel = output_channel
 
         # out
         num_groups_out = norm_num_groups if norm_num_groups is not None else min(block_out_channels[0] // 4, 32)

Original file line number	Diff line number	Diff line change
`@@ -240,7 +240,6 @@ def __init__(`
`240`	`240`	`dropout=dropout,`
`241`	`241`	`)`
`242`	`242`	`self.up_blocks.append(up_block)`
`243`		`- prev_output_channel = output_channel`
`244`	`243`
`245`	`244`	`# out`
`246`	`245`	`num_groups_out = norm_num_groups if norm_num_groups is not None else min(block_out_channels[0] // 4, 32)`