AI/diffusers
1
0
Fork 0
mirror of https://github.com/huggingface/diffusers.git synced 2026-01-27 17:22:53 +03:00
Code Activity

Fix vae.Decoder prev_output_channel (#11280)

Browse Source
This commit is contained in:
hlky
2025-04-15 08:03:50 +01:00
committed by GitHub
parent 9352a5ca56
commit 6e80d240d3
1 changed files with 1 additions and 1 deletions
Download Patch File Download Diff File Expand all files Collapse all files

2
src/diffusers/models/autoencoders/vae.py
View File

@@ -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,