We read every piece of feedback, and take your input very seriously.
To see all available qualifiers, see our documentation.
There was an error while loading. Please reload this page.
1 parent b8093e6 commit 2fad3abCopy full SHA for 2fad3ab
src/diffusers/models/autoencoders/vae.py
@@ -255,7 +255,7 @@ def __init__(
255
num_layers=self.layers_per_block + 1,
256
in_channels=prev_output_channel,
257
out_channels=output_channel,
258
- prev_output_channel=None,
+ prev_output_channel=prev_output_channel,
259
add_upsample=not is_final_block,
260
resnet_eps=1e-6,
261
resnet_act_fn=act_fn,
0 commit comments