Skip to content

fix(ltx): use the 2.3/22b distill LoRA in createLtxI2VWorkflow - #59

Merged
mystixxx merged 1 commit into
stagefrom
fix/ltx-distill-lora-2.3
Sep 7, 2026
Merged

mystixxx merged 1 commit into
stagefrom
fix/ltx-distill-lora-2.3

Conversation

@scottdraves

Copy link
Copy Markdown
Contributor

Node 7 of createLtxI2VWorkflow loaded ltx-2-19b-distilled-lora-384 onto the ltx-2.3-22b-dev transformer — a 19B LoRA on a 22B model.

It is 99% key-compatible, which is why this went unnoticed. Matching every key from the safetensors headers against the deployed transformer:

19B distill LoRA (currently deployed)
  modules: 1371 | bind OK: 1364 (99%) | MISSING: 5 | SHAPE mismatch: 2
    x adaln_single.linear        lora [24576, 4096]  model [36864, 4096]
    x audio_adaln_single.linear  lora [12288, 2048]  model [18432, 2048]

The two failures are the adaLN modulation layers that drive every block. ComfyUI logs a shape error and applies the other 1364 modules anyway → half-distilled model → degraded 8+3-step LCM output. This is the error on the running endpoint (12288 × 2048 = 25165824):

ERROR lora diffusion_model.audio_adaln_single.linear.weight
shape [18432, 2048] is invalid for input of size 25165824

It names the audio tensor, but the video adaln_single.linear fails identically — the symptom is visual.

Lightricks/LTX-2.3 ships the matching build: 1660/1660 modules bind, 0 missing, 0 mismatched.

The camera-control LoRAs are unaffected — all 480 of their modules already bind cleanly to the dev transformer (48 blocks, width 4096 both sides), and they carry no adaLN or audio tensors. The June rewiring (node 1 dev → node 7 distill → node 6 camera → both CFGGuiders) was correct; only the distill filename was wrong.

Requires the paired container change

The worker only names the LoRA; the file must exist in the image. Merge and release that first, or ComfyUI will fail with a missing-LoRA error instead:

e-dream-ai/gpu-container-ltx#2

Order: container → GHCR → manual RunPod New Release → this.

npx tsc --noEmit clean.

🤖 Generated with Claude Code

https://claude.ai/code/session_015Mwf3U49A8BXV8Rp2zQGNB

Node 7 loaded ltx-2-19b-distilled-lora-384 onto the ltx-2.3-22b-dev
transformer. Comparing safetensors headers, 1364 of that LoRA's 1371
modules bind to the 22b model, but the two that fail are the adaLN
modulation layers:

  adaln_single.linear        lora [24576, 4096]  model [36864, 4096]
  audio_adaln_single.linear  lora [12288, 2048]  model [18432, 2048]

ComfyUI logs a shape error for those (12288*2048 = 25165824, the size in
the reported error) and applies the remaining modules anyway. The result
is a half-distilled model: blocks adapted for few-step sampling driven by
unpatched modulation, which degrades the 8+3-step LCM output.

Lightricks/LTX-2.3 ships the matching build; all 1660 of its modules bind
to this transformer with no missing keys and no shape mismatches.

The camera-control LoRAs are unaffected — all 480 of their modules already
bind cleanly to the dev transformer (48 blocks, width 4096 both sides) and
they contain no adaLN or audio tensors.

Requires the paired gpu-container-ltx change that bakes the new file.

Co-Authored-By: Claude Opus 5 (1M context) <noreply@anthropic.com>
Claude-Session: https://claude.ai/code/session_015Mwf3U49A8BXV8Rp2zQGNB
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment

Labels

None yet

Projects

None yet

Development

Successfully merging this pull request may close these issues.

2 participants