Skip to content

GitLab

Unverified Commit 6e80d240 authored by hlky's avatar hlky Committed by GitHub
Browse files

Fix vae.Decoder prev_output_channel (#11280)

parent 9352a5ca
Hide whitespace changes
Inline Side-by-side
Showing with 1 addition and 1 deletion
src/diffusers/models/autoencoders/vae.py
View file @ 6e80d240
...@@ -255,7 +255,7 @@ class Decoder(nn.Module): ...@@ -255,7 +255,7 @@ class Decoder(nn.Module):
num_layers=self.layers_per_block + 1, num_layers=self.layers_per_block + 1,
in_channels=prev_output_channel, in_channels=prev_output_channel,
out_channels=output_channel, out_channels=output_channel,
prev_output_channel=None, prev_output_channel=prev_output_channel,
add_upsample=not is_final_block, add_upsample=not is_final_block,
resnet_eps=1e-6, resnet_eps=1e-6,
resnet_act_fn=act_fn, resnet_act_fn=act_fn,
......
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