We read every piece of feedback, and take your input very seriously.
To see all available qualifiers, see our documentation.
There was an error while loading. Please reload this page.
1 parent 9ef7b59 commit 074817cCopy full SHA for 074817c
scripts/convert_dcae_to_diffusers.py
@@ -97,7 +97,7 @@ def convert_ae(ckpt_path: str, dtype: torch.dtype):
97
decoder_layers_per_block=(3, 3, 3, 3, 3, 3),
98
encoder_qkv_multiscales=((), (), (), (5,), (5,), (5,)),
99
decoder_qkv_multiscales=((), (), (), (5,), (5,), (5,)),
100
- downsample_block_type="Conv",
+ downsample_block_type="conv",
101
upsample_block_type="interpolate",
102
decoder_norm_types="rms_norm",
103
decoder_act_fns="silu",
@@ -131,7 +131,7 @@ def get_ae_config(name: str):
131
"decoder_qkv_multiscales": ((), (), (), (5,), (5,), (5,)),
132
"encoder_layers_per_block": (2, 2, 2, 3, 3, 3),
133
"decoder_layers_per_block": [3, 3, 3, 3, 3, 3],
134
- "downsample_block_type": "Conv",
+ "downsample_block_type": "conv",
135
"upsample_block_type": "interpolate",
136
"scaling_factor": 0.41407,
137
}
0 commit comments