Skip to content

Commit a7b9157

Browse files
committed
fix peft delete adapters for flux.
1 parent ba8dc7d commit a7b9157

File tree

1 file changed

+1
-2
lines changed

1 file changed

+1
-2
lines changed

src/diffusers/loaders/lora_pipeline.py

Lines changed: 1 addition & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -2545,14 +2545,13 @@ def _maybe_expand_lora_state_dict(cls, transformer, lora_state_dict):
25452545
if unexpected_modules:
25462546
logger.debug(f"Found unexpected modules: {unexpected_modules}. These will be ignored.")
25472547

2548-
is_peft_loaded = getattr(transformer, "peft_config", None) is not None
25492548
for k in lora_module_names:
25502549
if k in unexpected_modules:
25512550
continue
25522551

25532552
base_param_name = (
25542553
f"{k.replace(prefix, '')}.base_layer.weight"
2555-
if is_peft_loaded and f"{k.replace(prefix, '')}.base_layer.weight" in transformer_state_dict
2554+
if f"{k.replace(prefix, '')}.base_layer.weight" in transformer_state_dict
25562555
else f"{k.replace(prefix, '')}.weight"
25572556
)
25582557
base_weight_param = transformer_state_dict[base_param_name]

0 commit comments

Comments
 (0)