Fix

Author: Perseus14

src/maxdiffusion/models/lora_nnx.py

@@ -302,10 +302,10 @@ def merge_lora(model: nnx.Module, state_dict: dict, scale: float):
                       else:
                         raise NotImplementedError(f"Conv merge only for 1x1 kernels, got {module.kernel_size}")
             else:
-                max_logging.warning(f"LoRA weights for {lora_key} incomplete: missing down or up weights.")
+                max_logging.log(f"LoRA weights for {lora_key} incomplete: missing down or up weights.")
         elif lora_key:
-              max_logging.warning(f"NNX layer '{nnx_path_str}' translated to '{lora_key}' but key not in lora_params.")
+              max_logging.log(f"NNX layer '{nnx_path_str}' translated to '{lora_key}' but key not in lora_params.")
         else:
-            max_logging.debug(f"NNX layer '{nnx_path_str}' could not be translated to a LoRA key.")
+            max_logging.log(f"NNX layer '{nnx_path_str}' could not be translated to a LoRA key.")
 
     max_logging.log(f"Merged weights into {assigned_count} layers in {type(model).__name__}.")