From fc0eddb9fd792776ad98e8143baab196a9c9881f Mon Sep 17 00:00:00 2001 From: Ahmed Belgacem Date: Mon, 3 Mar 2025 15:38:24 +0100 Subject: [PATCH] Fix redundant prev_output_channel assignment in UNet2DModel --- src/diffusers/models/unets/unet_2d.py | 1 - 1 file changed, 1 deletion(-) diff --git a/src/diffusers/models/unets/unet_2d.py b/src/diffusers/models/unets/unet_2d.py index 5a7fc32223d6..448ec051a032 100644 --- 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)