Skip to content

Commit 9290a6e

Browse files
committed
Fix conversion logic
1 parent 1b0976d commit 9290a6e

1 file changed

Lines changed: 1 addition & 1 deletion

File tree

src/maxdiffusion/loaders/lora_conversion_utils.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -631,7 +631,7 @@ def translate_wan_nnx_path_to_diffusers_lora(nnx_path_str, scan_layers=False):
631631
return 'diffusion_model.img_emb.proj.1'
632632
if nnx_path_str == 'condition_embedder.image_embedder.ff.net_2':
633633
return 'diffusion_model.img_emb.proj.3'
634-
if nnx_path_str == 'condition_embedder.image_embedder.norm2':
634+
if nnx_path_str == 'condition_embedder.image_embedder.norm2.layer_norm':
635635
return 'diffusion_model.img_emb.proj.4'
636636
if nnx_path_str == 'patch_embedding':
637637
return 'diffusion_model.patch_embedding'

0 commit comments

Comments
 (0)