Skip to content

LTX2 Upsampler fix#378

Open
prishajain1 wants to merge 1 commit intomainfrom
upsampler_correction
Open

LTX2 Upsampler fix#378
prishajain1 wants to merge 1 commit intomainfrom
upsampler_correction

Conversation

@prishajain1
Copy link
Copy Markdown
Collaborator

The current LTX2 Upsampler pipeline raises ValueError : blur_down is the name of the submodule in the PyTorch state dict from the Hugging Face checkpoint.

In the original PyTorch model, that layer was named blur_down, but in the MaxDiffusion Flax implementation, it was named blur. Because our weight loader didn't rename it, nnx.update tried to update a non-existent blur_down attribute.

This PR fixes the above issue

@prishajain1 prishajain1 requested a review from entrpn as a code owner April 15, 2026 13:20
@github-actions
Copy link
Copy Markdown

Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment

Projects

None yet

Development

Successfully merging this pull request may close these issues.

2 participants