From b34df2061ec7ac91249fefd0ab5103b19048cb14 Mon Sep 17 00:00:00 2001 From: Daniel Gu Date: Wed, 12 Nov 2025 06:22:15 +0100 Subject: [PATCH] Revert dim_mult back to list and fix type annotation --- src/diffusers/models/autoencoders/autoencoder_kl_wan.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/src/diffusers/models/autoencoders/autoencoder_kl_wan.py b/src/diffusers/models/autoencoders/autoencoder_kl_wan.py index 5b4b74543ae3..b0b2960aaf18 100644 --- a/src/diffusers/models/autoencoders/autoencoder_kl_wan.py +++ b/src/diffusers/models/autoencoders/autoencoder_kl_wan.py @@ -971,7 +971,7 @@ def __init__( base_dim: int = 96, decoder_base_dim: Optional[int] = None, z_dim: int = 16, - dim_mult: Tuple[int, ...] = (1, 2, 4, 4), + dim_mult: List[int] = [1, 2, 4, 4], num_res_blocks: int = 2, attn_scales: List[float] = [], temperal_downsample: List[bool] = [False, True, True],