test

Author: Perseus14

src/maxdiffusion/loaders/lora_conversion_utils.py

@@ -651,6 +651,8 @@ def translate_wan_nnx_path_to_diffusers_lora(nnx_path_str, scan_layers=False):
     return "diffusion_model.patch_embedding"
   if nnx_path_str == "proj_out":
     return "diffusion_model.head.head"
+  if nnx_path_str == "adaln_scale_shift_table":
+    return "diffusion_model.head.modulation"
   if nnx_path_str == "condition_embedder.time_proj":
     return "diffusion_model.time_projection.1"
 

src/maxdiffusion/models/lora_nnx.py

@@ -300,7 +300,6 @@ def merge_lora_for_scanned(
       max_logging.log(f"lora_key_template: {lora_key_template}")
       max_logging.log(f"Module ndim: {module.ndim}")
       max_logging.log(f"Module Type: {type(module)}")
-      continue
 
     if not lora_key_template:
       continue