Fix

Author: Perseus14

src/maxdiffusion/configs/base_wan_lora_14b.yml

@@ -296,7 +296,7 @@ lora_rank: 64
 lora_config: {
   lora_model_name_or_path: ["lightx2v/Wan2.1-Distill-Loras"],
   weight_name: ["Wan2.1_t2v_14b_lora_rank64_lightx2v_4step.safetensors"],
-  adapter_name: ["wan21-lora"],
+  adapter_name: ["wan21-distill-lora"],
   scale: [1.0],
   from_pt: []
 }

src/maxdiffusion/configs/base_wan_lora_27b.yml

@@ -305,10 +305,10 @@ lightning_ckpt: ""
 lora_rank: 64
 # Values are lists to support multiple LoRA loading during inference in the future.
 lora_config: {
-  lora_model_name_or_path: ["lightx2v/Wan2.2-Lightning"],
-  high_noise_weight_name: ["Wan2.2-T2V-A14B-4steps-lora-250928/high_noise_model.safetensors"],
-  low_noise_weight_name: ["Wan2.2-T2V-A14B-4steps-lora-250928/low_noise_model.safetensors"],
-  adapter_name: ["wan22-lightning-lora"],
+  lora_model_name_or_path: ["lightx2v/Wan2.2-Distill-Loras"],
+  high_noise_weight_name: ["wan2.2_t2v_A14b_high_noise_lora_rank64_lightx2v_4step_1217.safetensors"],
+  low_noise_weight_name: ["wan2.2_t2v_A14b_low_noise_lora_rank64_lightx2v_4step_1217.safetensors"],
+  adapter_name: ["wan22-distill-lora"],
   scale: [1.0],
   from_pt: []
 }

src/maxdiffusion/models/lora_nnx.py

@@ -349,7 +349,7 @@ def merge_lora_for_scanned(model: nnx.Module, state_dict: dict, scale: float, tr
         # Handle scanned Conv layers (ndim=5)
         elif isinstance(module, nnx.Conv) and module.kernel.ndim == 5:
             if module.kernel_size != (1, 1):
-                max_logging.warn(f"Skipping merge for scanned Conv layer {nnx_path_str} with kernel size {module.kernel_size}, only 1x1 is supported for merging.")
+                max_logging.log(f"Skipping merge for scanned Conv layer {nnx_path_str} with kernel size {module.kernel_size}, only 1x1 is supported for merging.")
                 continue
 
             lora_key_template = translate_fn(nnx_path_str) if translate_fn else None