Skip to content

Commit 1b0976d

Browse files
committed
Fix conversion logic
1 parent fc0e280 commit 1b0976d

1 file changed

Lines changed: 3 additions & 3 deletions

File tree

src/maxdiffusion/loaders/lora_conversion_utils.py

Lines changed: 3 additions & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -625,11 +625,11 @@ def translate_wan_nnx_path_to_diffusers_lora(nnx_path_str, scan_layers=False):
625625
return 'diffusion_model.time_embedding.0'
626626
if nnx_path_str == 'condition_embedder.time_embedder.linear_2':
627627
return 'diffusion_model.time_embedding.2'
628-
if nnx_path_str == 'condition_embedder.image_embedder.norm1':
628+
if nnx_path_str == 'condition_embedder.image_embedder.norm1.layer_norm':
629629
return 'diffusion_model.img_emb.proj.0'
630-
if nnx_path_str == 'condition_embedder.image_embedder.ff.net.0.proj':
630+
if nnx_path_str == 'condition_embedder.image_embedder.ff.net_0':
631631
return 'diffusion_model.img_emb.proj.1'
632-
if nnx_path_str == 'condition_embedder.image_embedder.ff.net.2':
632+
if nnx_path_str == 'condition_embedder.image_embedder.ff.net_2':
633633
return 'diffusion_model.img_emb.proj.3'
634634
if nnx_path_str == 'condition_embedder.image_embedder.norm2':
635635
return 'diffusion_model.img_emb.proj.4'

0 commit comments

Comments
 (0)