Fix

Author: Perseus14

src/maxdiffusion/models/lora_nnx.py

@@ -222,7 +222,7 @@ def merge_lora(model: nnx.Module, state_dict: dict, scale: float):
               if m:
                   module_path_str, weight_type = m.group(1), m.group(2).replace("lora_", "")
               else:
-                  max_logging.debug(f"Could not parse LoRA key: {k}")
+                  max_logging.log(f"Could not parse LoRA key: {k}")
                   continue
 
       if module_path_str not in lora_params:
@@ -267,5 +267,5 @@ def merge_lora(model: nnx.Module, state_dict: dict, scale: float):
                           f"with kernel_size {module.kernel_size} > 1 is not supported."
                       )
           else:
-              max_logging.warning(f"LoRA weights for {matched_key} incomplete.")
+              max_logging.log(f"LoRA weights for {matched_key} incomplete.")
     max_logging.log(f"Merged weights into {assigned_count} layers in {type(model).__name__}.")