diff --git a/src/diffusers/models/autoencoders/vae.py b/src/diffusers/models/autoencoders/vae.py index 826a6365b7..75503dc6a4 100644 --- a/src/diffusers/models/autoencoders/vae.py +++ b/src/diffusers/models/autoencoders/vae.py @@ -90,7 +90,6 @@ class Encoder(nn.Module): padding=1, ) - self.mid_block = None self.down_blocks = nn.ModuleList([]) # down @@ -228,7 +227,6 @@ class Decoder(nn.Module): padding=1, ) - self.mid_block = None self.up_blocks = nn.ModuleList([]) temb_channels = in_channels if norm_type == "spatial" else None @@ -474,7 +472,6 @@ class MaskConditionDecoder(nn.Module): padding=1, ) - self.mid_block = None self.up_blocks = nn.ModuleList([]) temb_channels = in_channels if norm_type == "spatial" else None