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

Revert AutoencoderKLWan's dim_mult default value back to list (#12640)

Browse Source 
Revert dim_mult back to list and fix type annotation
This commit is contained in:
dg845
2025-11-17 05:09:53 -08:00
committed by GitHub
parent 3579fdabf9
commit 67dc65e2e3
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/autoencoder_kl_wan.py
View File

@@ -971,7 +971,7 @@ class AutoencoderKLWan(ModelMixin, AutoencoderMixin, ConfigMixin, FromOriginalMo
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],