Skip to content

Commit 58559ec

Browse files
committed
no need modify as peft updated
1 parent 48eeeae commit 58559ec

File tree

1 file changed

+1
-1
lines changed

1 file changed

+1
-1
lines changed

src/diffusers/utils/peft_utils.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -182,7 +182,7 @@ def get_peft_kwargs(
182182
else:
183183
lora_alpha = set(network_alpha_dict.values()).pop()
184184

185-
target_modules = list({name.split(".lora")[0] for name in peft_state_dict.keys() if "lora" in name})
185+
target_modules = list({name.split(".lora")[0] for name in peft_state_dict.keys()})
186186
use_dora = any("lora_magnitude_vector" in k for k in peft_state_dict)
187187
# for now we know that the "bias" keys are only associated with `lora_B`.
188188
lora_bias = any("lora_B" in k and k.endswith(".bias") for k in peft_state_dict)

0 commit comments

Comments
 (0)