From b0ee2b53cbc062e19df55c02a49e8e9f862c47a6 Mon Sep 17 00:00:00 2001 From: Kartavya Dikshit Date: Mon, 22 Jun 2026 10:20:19 +0200 Subject: [PATCH] fix: handle CLIP text_model prefix change in transformers>=5 for FLUX LoRA Signed-off-by: Kartavya Dikshit --- src/diffusers/loaders/lora_base.py | 6 ++++++ 1 file changed, 6 insertions(+) diff --git a/src/diffusers/loaders/lora_base.py b/src/diffusers/loaders/lora_base.py index 5b5579664b55..12b50076fb33 100644 --- a/src/diffusers/loaders/lora_base.py +++ b/src/diffusers/loaders/lora_base.py @@ -381,6 +381,12 @@ def _load_lora_into_text_encoder( rank_key = f"{name}.lora_B.weight" if rank_key in state_dict: rank[rank_key] = state_dict[rank_key].shape[1] + else: + # Handle transformers>=5 where CLIPTextModel was flattened and no longer + # includes the `text_model.` wrapper prefix in named_modules(). + alt_rank_key = f"text_model.{name}.lora_B.weight" + if alt_rank_key in state_dict: + rank[alt_rank_key] = state_dict[alt_rank_key].shape[1] if network_alphas is not None: alpha_keys = [k for k in network_alphas.keys() if k.startswith(prefix) and k.split(".")[0] == prefix]