We read every piece of feedback, and take your input very seriously.
To see all available qualifiers, see our documentation.
1 parent 039e688 commit 9884ed2Copy full SHA for 9884ed2
src/diffusers/models/autoencoders/autoencoder_kl_flux2.py
@@ -91,6 +91,7 @@ def __init__(
91
512,
92
93
),
94
+ decoder_block_out_channels: tuple[int, ...] | None = None,
95
layers_per_block: int = 2,
96
act_fn: str = "silu",
97
latent_channels: int = 32,
@@ -124,7 +125,7 @@ def __init__(
124
125
in_channels=latent_channels,
126
out_channels=out_channels,
127
up_block_types=up_block_types,
- block_out_channels=block_out_channels,
128
+ block_out_channels=decoder_block_out_channels or block_out_channels,
129
layers_per_block=layers_per_block,
130
norm_num_groups=norm_num_groups,
131
act_fn=act_fn,
0 commit comments