Skip to content

Commit 6e80d24

Browse files
authored
Fix vae.Decoder prev_output_channel (huggingface#11280)
1 parent 9352a5c commit 6e80d24

File tree

1 file changed

+1
-1
lines changed
  • src/diffusers/models/autoencoders

1 file changed

+1
-1
lines changed

src/diffusers/models/autoencoders/vae.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -255,7 +255,7 @@ def __init__(
255255
num_layers=self.layers_per_block + 1,
256256
in_channels=prev_output_channel,
257257
out_channels=output_channel,
258-
prev_output_channel=None,
258+
prev_output_channel=prev_output_channel,
259259
add_upsample=not is_final_block,
260260
resnet_eps=1e-6,
261261
resnet_act_fn=act_fn,

0 commit comments

Comments
 (0)