Skip to content

Conversation

@sayakpaul
Copy link
Member

What does this PR do?

Fix #13107

Cc: @mshehzadkhan. Could you give this a try please?

@sayakpaul sayakpaul requested a review from dg845 February 10, 2026 15:40
@HuggingFaceDocBuilderDev

The docs for this PR live here. All of your documentation changes will be reflected on that endpoint. The docs are available until 30 days after the last update.

@mshehzadkhan
Copy link

mshehzadkhan commented Feb 10, 2026

I’ve tested a bunch of Klein LoRAs for each model from CivitAI, and they all seem to be loading correctly now.

Thanks a lot @sayakpaul for the quick fix 🙌

Copy link
Collaborator

@dg845 dg845 left a comment

Choose a reason for hiding this comment

The reason will be displayed to describe this comment to others. Learn more.

Thanks! If I understand correctly, the changes can be summarized as follows:

  1. Support LoRAs which name the A and B matrices lora_down/lora_up instead of lora_A/lora_B respectively.
  2. Support LoRAs which only target a subset of the transformer block weights (e.g. a subset of the attn QKV projections, or attn out projections, etc.)
  3. Support LoRAs which target weights outside of the transformer blocks, as defined by extra_mappings.

@sayakpaul sayakpaul merged commit 4d00980 into main Feb 11, 2026
11 of 12 checks passed
@sayakpaul sayakpaul deleted the fix-flux2-lora-keys branch February 11, 2026 02:37
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.

Flux Klein LoRA – Missing Key Errors

4 participants