Skip to content

Commit 2fad3ab

Browse files
committed
Fix vae.Decoder prev_output_channel
1 parent b8093e6 commit 2fad3ab

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)