comments

Author: tsvikas

src/maxdiffusion/configs/ltx2_3_video.yml

@@ -145,6 +145,8 @@ enable_lora: False
 #   lora_model_name_or_path: ["Lightricks/LTX-2.3"],
 #   weight_name: ["ltx-2.3-22b-distilled-lora-384.safetensors"],
 #   adapter_name: ["distilled-lora-384"],
+#   # placeholder - the real value is mixed per-layer ranks: 32/128/256/384
+#   # and the loader reads each layer's REAL rank from the LoRA tensor shapes
 #   rank: [384]
 # }
 

src/maxdiffusion/loaders/lora_conversion_utils.py

@@ -716,7 +716,7 @@ def translate_ltx2_nnx_path_to_diffusers_lora(nnx_path_str, scan_layers=False):
       "attn1.to_k": "attn1.to_k",
       "attn1.to_v": "attn1.to_v",
       "attn1.to_out": "attn1.to_out.0",
-      "attn1.to_gate_logits": "attn1.to_gate_logits",  # per-head gating Linear
+      "attn1.to_gate_logits": "attn1.to_gate_logits",
       # Audio Self Attention (audio_attn1)
       "audio_attn1.to_q": "audio_attn1.to_q",
       "audio_attn1.to_k": "audio_attn1.to_k",