Skip to content

GitLab

Unverified Commit 8f15be16 authored by Ahmed Belgacem's avatar Ahmed Belgacem Committed by GitHub
Browse files

Fix redundant prev_output_channel assignment in UNet2DModel (#10945)

parent f92e599c
Hide whitespace changes
Inline Side-by-side
Showing with 0 additions and 1 deletion
src/diffusers/models/unets/unet_2d.py
View file @ 8f15be16
...@@ -240,7 +240,6 @@ class UNet2DModel(ModelMixin, ConfigMixin): ...@@ -240,7 +240,6 @@ class UNet2DModel(ModelMixin, ConfigMixin):
dropout=dropout, dropout=dropout,
) )
self.up_blocks.append(up_block) self.up_blocks.append(up_block)
prev_output_channel = output_channel
# out # out
num_groups_out = norm_num_groups if norm_num_groups is not None else min(block_out_channels[0] // 4, 32) num_groups_out = norm_num_groups if norm_num_groups is not None else min(block_out_channels[0] // 4, 32)
......
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