Skip to content

Commit 074817c

Browse files
committed
replace downsample_block_type from Conv to conv for consistency
1 parent 9ef7b59 commit 074817c

File tree

1 file changed

+2
-2
lines changed

1 file changed

+2
-2
lines changed

scripts/convert_dcae_to_diffusers.py

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -97,7 +97,7 @@ def convert_ae(ckpt_path: str, dtype: torch.dtype):
9797
decoder_layers_per_block=(3, 3, 3, 3, 3, 3),
9898
encoder_qkv_multiscales=((), (), (), (5,), (5,), (5,)),
9999
decoder_qkv_multiscales=((), (), (), (5,), (5,), (5,)),
100-
downsample_block_type="Conv",
100+
downsample_block_type="conv",
101101
upsample_block_type="interpolate",
102102
decoder_norm_types="rms_norm",
103103
decoder_act_fns="silu",
@@ -131,7 +131,7 @@ def get_ae_config(name: str):
131131
"decoder_qkv_multiscales": ((), (), (), (5,), (5,), (5,)),
132132
"encoder_layers_per_block": (2, 2, 2, 3, 3, 3),
133133
"decoder_layers_per_block": [3, 3, 3, 3, 3, 3],
134-
"downsample_block_type": "Conv",
134+
"downsample_block_type": "conv",
135135
"upsample_block_type": "interpolate",
136136
"scaling_factor": 0.41407,
137137
}

0 commit comments

Comments
 (0)