Skip to content

GitLab

"git@developer.sourcefind.cn:norm/vllm.git" did not exist on "56b7f0efa4d83865afc2da38b40b2f337d778dda"
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):
num_layers=self.layers_per_block + 1,
in_channels=prev_output_channel,
out_channels=output_channel,
prev_output_channel=None,
prev_output_channel=prev_output_channel,
add_upsample=not is_final_block,
resnet_eps=1e-6,
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