From 3325491eabda6fbe6f0f431be332ccc96c3b7d21 Mon Sep 17 00:00:00 2001 From: FeiXie8 <946085309@qq.com> Date: Fri, 17 Oct 2025 22:15:08 +0800 Subject: [PATCH] Fix: Use incorrect temporary variable key when replacing adapter name in state dict within load_lora_adapter function --- src/diffusers/loaders/peft.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/src/diffusers/loaders/peft.py b/src/diffusers/loaders/peft.py index 2381ccfef34a..7d65b30659fb 100644 --- a/src/diffusers/loaders/peft.py +++ b/src/diffusers/loaders/peft.py @@ -293,7 +293,7 @@ def map_state_dict_for_hotswap(sd): # For hotswapping, we need the adapter name to be present in the state dict keys new_sd = {} for k, v in sd.items(): - if k.endswith("lora_A.weight") or key.endswith("lora_B.weight"): + if k.endswith("lora_A.weight") or k.endswith("lora_B.weight"): k = k[: -len(".weight")] + f".{adapter_name}.weight" elif k.endswith("lora_B.bias"): # lora_bias=True option k = k[: -len(".bias")] + f".{adapter_name}.bias"