Add comment

Author: Perseus14

src/maxdiffusion/models/lora_nnx.py

@@ -91,8 +91,9 @@ def parse_lora_dict(state_dict, dtype):
   """
   Helper to parse state_dict into structured params including diffs.
   Supports keys in original WAN LoRA format, like:
-  '...lora.down.weight', '...lora_up.weight', '...alpha',
+  '...lora_A.weight', '...lora_up.weight', '...alpha',
   as well as '.diff' and '.diff_b' for weight and bias fine-tuning.
+  Supports ComfyUI and AI toolkit lora formats.
   """
   lora_params = {}
   for k, v in state_dict.items():