Simpler way to implement the #11980 loras. (#11981)

2026-01-26 19:19:53 +00:00 · 2026-01-19 19:00:36 -08:00
parent 7458e20465
commit e0eacb0688
1 changed files with 2 additions and 0 deletions
--- a/comfy/utils.py
+++ b/comfy/utils.py
@@ -639,6 +639,8 @@ def flux_to_diffusers(mmdit_config, output_prefix=""):
                        "proj_out.bias": "linear2.bias",
                        "attn.norm_q.weight": "norm.query_norm.scale",
                        "attn.norm_k.weight": "norm.key_norm.scale",
+                        "attn.to_qkv_mlp_proj.weight": "linear1.weight", # Flux 2
+                        "attn.to_out.weight": "linear2.weight", # Flux 2
                    }

        for k in block_map: