Fix

Author: Perseus14

src/maxdiffusion/models/lora_nnx.py

@@ -227,6 +227,7 @@ def merge_lora(model: nnx.Module, state_dict: dict, scale: float, translate_fn=N
         if not isinstance(module, (nnx.Linear, nnx.Conv)): continue
 
         nnx_path_str = ".".join(map(str, path))
+        max_logging.log(f"NNX path: {nnx_path_str}")
         lora_key = translate_fn(nnx_path_str) if translate_fn else None
 
         if lora_key and lora_key in lora_params: