InvokeAI/invokeai/backend/model_manager/configs
Lincoln Stein dfc66b7142
Feature: Add FLUX.2 LOKR model support (detection and loading) (#8909)
* Add FLUX.2 LOKR model support (detection and loading) (#88)

Fix BFL LOKR models being misidentified as AIToolkit format



Fix alpha key warning in LOKR QKV split layers

Co-authored-by: copilot-swe-agent[bot] <198982749+Copilot@users.noreply.github.com>
Co-authored-by: lstein <111189+lstein@users.noreply.github.com>

* Fix BFL→diffusers key mapping for non-block layers in FLUX.2 LoRA/LoKR

BFL's FLUX.2 model uses different names than diffusers' Flux2Transformer2DModel
for top-level modules (embedders, modulations, output layers). The existing
conversion only handled block-level renames (double_blocks→transformer_blocks),
causing "Failed to find module" warnings for non-block LoRA keys like img_in,
txt_in, modulation.lin, time_in, and final_layer.

---------

Co-authored-by: Copilot <198982749+Copilot@users.noreply.github.com>
Co-authored-by: lstein <111189+lstein@users.noreply.github.com>
Co-authored-by: Alexander Eichhorn <alex@eichhorn.dev>
2026-02-27 00:45:13 +00:00
..
__init__.py
base.py feat(z-image): add Z-Image Base (undistilled) model variant support (#8799) 2026-02-20 00:32:38 +00:00
clip_embed.py Feature(backend): Add user toggle to run encoder models on CPU (#8777) 2026-02-04 15:13:29 -05:00
clip_vision.py Feature(backend): Add user toggle to run encoder models on CPU (#8777) 2026-02-04 15:13:29 -05:00
controlnet.py WIP not working. 2025-12-21 18:43:02 +01:00
external_api.py
factory.py Feature: flux2 klein lora support (#8862) 2026-02-24 02:55:39 +00:00
flux_redux.py
identification_utils.py
ip_adapter.py
llava_onevision.py Feature(backend): Add user toggle to run encoder models on CPU (#8777) 2026-02-04 15:13:29 -05:00
lora.py Feature: Add FLUX.2 LOKR model support (detection and loading) (#8909) 2026-02-27 00:45:13 +00:00
main.py feat(z-image): add Z-Image Base (undistilled) model variant support (#8799) 2026-02-20 00:32:38 +00:00
qwen3_encoder.py Feature(backend): Add user toggle to run encoder models on CPU (#8777) 2026-02-04 15:13:29 -05:00
siglip.py Feature(backend): Add user toggle to run encoder models on CPU (#8777) 2026-02-04 15:13:29 -05:00
spandrel.py
t2i_adapter.py
t5_encoder.py Feature(backend): Add user toggle to run encoder models on CPU (#8777) 2026-02-04 15:13:29 -05:00
textual_inversion.py
unknown.py
vae.py fix(model_manager): detect Flux1/2 VAE by latent space dimensions instead of filename (#8790) 2026-01-27 05:20:50 +00:00