Skip to content

Commit 5d45b0c

Browse files
committed
update
1 parent 6e2a93d commit 5d45b0c

File tree

4 files changed

+330
-48
lines changed

4 files changed

+330
-48
lines changed

src/diffusers/loaders/single_file_model.py

Lines changed: 10 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -39,6 +39,8 @@
3939
convert_mochi_transformer_checkpoint_to_diffusers,
4040
convert_sd3_transformer_checkpoint_to_diffusers,
4141
convert_stable_cascade_unet_single_file_to_diffusers,
42+
convert_wan_transformer_to_diffusers,
43+
convert_wan_vae_to_diffusers,
4244
create_controlnet_diffusers_config_from_ldm,
4345
create_unet_diffusers_config_from_ldm,
4446
create_vae_diffusers_config_from_ldm,
@@ -117,6 +119,14 @@
117119
"checkpoint_mapping_fn": convert_lumina2_to_diffusers,
118120
"default_subfolder": "transformer",
119121
},
122+
"WanTransformer3DModel": {
123+
"checkpoint_mapping_fn": convert_wan_transformer_to_diffusers,
124+
"default_subfolder": "transformer",
125+
},
126+
"AutoencoderKLWan": {
127+
"checkpoint_mapping_fn": convert_wan_vae_to_diffusers,
128+
"default_subfolder": "vae",
129+
},
120130
}
121131

122132

0 commit comments

Comments
 (0)