Skip to content

Fix LoKr format LoRA weight loading for Flux2 pipelines#13997

Open
Whning0513 wants to merge 1 commit into
huggingface:mainfrom
Whning0513:fix-issue-13261
Open

Fix LoKr format LoRA weight loading for Flux2 pipelines#13997
Whning0513 wants to merge 1 commit into
huggingface:mainfrom
Whning0513:fix-issue-13261

Conversation

@Whning0513

@Whning0513 Whning0513 commented Jun 19, 2026

Copy link
Copy Markdown

LoKr weights were failing to load with a ValueError about non-empty original_state_dict. Adjusted the weight processing to handle the LoKr format correctly.

@github-actions github-actions Bot added lora size/S PR with diff < 50 LOC fixes-issue labels Jun 19, 2026
@Whning0513 Whning0513 changed the title Fix #13261 Fix LoKr format LoRA weight loading for Flux2 pipelines Jun 19, 2026
@github-actions

Copy link
Copy Markdown
Contributor

Hi @Whning0513, thanks for the PR! It does not appear to link an issue it fixes. If this PR addresses an existing issue, please add a closing keyword (e.g. Fixes #1234) to the PR description so the issue is linked. See the contribution guide for more details. If this PR intentionally does not fix a tracked issue, a maintainer can add the no-issue-needed label to silence this reminder.

@asomoza

asomoza commented Jun 19, 2026

Copy link
Copy Markdown
Member

Hi @Whning0513 this is not the correct way to handle lokr, with this fix the lora will load but practically won't have any effect.

Please look at this issue and this PR if you want to contribute

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

Labels

fixes-issue lora size/S PR with diff < 50 LOC

Projects

None yet

Development

Successfully merging this pull request may close these issues.

2 participants