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

This commit is contained in:
comfyanonymous 2026-01-19 19:00:36 -08:00 committed by GitHub
parent 7458e20465
commit e0eacb0688
No known key found for this signature in database
GPG Key ID: B5690EEEBB952194

View File

@ -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: