fix(lora): load ai-toolkit DoRA magnitudes across architectures - #9568
Open
zjn20030811 wants to merge 1 commit into
Open
fix(lora): load ai-toolkit DoRA magnitudes across architectures#9568zjn20030811 wants to merge 1 commit into
zjn20030811 wants to merge 1 commit into
Conversation
zjn20030811
requested review from
JPPhoto,
Pfannkuchensack,
blessedcoolant,
dunkeroni and
lstein
as code owners
September 3, 2026 16:08
This file contains hidden or bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
Sign up for free
to join this conversation on GitHub.
Already have an account?
Sign in to comment
Add this suggestion to a batch that can be applied as a single commit.This suggestion is invalid because no changes were made to the code.Suggestions cannot be applied while the pull request is closed.Suggestions cannot be applied while viewing a subset of changes.Only one suggestion per line can be applied in a batch.Add this suggestion to a batch that can be applied as a single commit.Applying suggestions on deleted lines is not supported.You must change the existing code in this line in order to create a valid suggestion.Outdated suggestions cannot be applied.This suggestion has been applied or marked resolved.Suggestions cannot be applied from pending reviews.Suggestions cannot be applied on multi-line comments.Suggestions cannot be applied while the pull request is queued to merge.Suggestion cannot be applied right now. Please check back later.
Summary
ai-toolkit DoRA adapters store the output-dimension magnitude as either
.magnitudeor.lora_magnitude_vector.weight. The LoRA converters previously split these keys with a fallback parser, producing invalid layer groups or an unsupported format error. This change recognizes both suffixes across the Wan, Z-Image, Anima, Qwen Image, and FLUX conversion paths and maps them to the existingdora_magnituderepresentation.The FLUX fused-QKV path now carries the magnitude (and existing input-dimension scale) into each output slice, preserving the correct orientation during conversion. Anima LoKR combinations strip either DoRA magnitude convention consistently.
Related Issues / Discussions
Closes #9558
QA Instructions
ruff check,ruff format --check, andgit diff --checkon all changed files.Merge Plan
Normal backend merge; no schema or database changes.