diff --git a/src/diffusers/loaders/lora_conversion_utils.py b/src/diffusers/loaders/lora_conversion_utils.py index f56632ced819..8b0f95b905e4 100644 --- a/src/diffusers/loaders/lora_conversion_utils.py +++ b/src/diffusers/loaders/lora_conversion_utils.py @@ -856,7 +856,7 @@ def _convert(original_key, diffusers_key, state_dict, new_state_dict): ) state_dict = {k: v for k, v in state_dict.items() if not k.startswith("text_encoders.t5xxl.transformer.")} - has_diffb = any("diff_b" in k and k.startswith(("lora_unet_", "lora_te_")) for k in state_dict) + has_diffb = any("diff_b" in k and k.startswith(("lora_unet_", "lora_te_", "lora_te1_")) for k in state_dict) if has_diffb: zero_status_diff_b = state_dict_all_zero(state_dict, ".diff_b") if zero_status_diff_b: @@ -895,7 +895,7 @@ def _convert(original_key, diffusers_key, state_dict, new_state_dict): state_dict = { _custom_replace(k, limit_substrings): v for k, v in state_dict.items() - if k.startswith(("lora_unet_", "lora_te_")) + if k.startswith(("lora_unet_", "lora_te_", "lora_te1_")) } if any("text_projection" in k for k in state_dict):