Skip to content

Commit 380398f

Browse files
zhaowendao30sayakpaulyiyixuxu
authored andcommitted
Update lora_conversion_utils.py (huggingface#9980)
x-flux single-blocks lora load Co-authored-by: Sayak Paul <spsayakpaul@gmail.com> Co-authored-by: YiYi Xu <yixu310@gmail.com>
1 parent c54b851 commit 380398f

File tree

1 file changed

+5
-1
lines changed

1 file changed

+5
-1
lines changed

src/diffusers/loaders/lora_conversion_utils.py

Lines changed: 5 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -643,7 +643,11 @@ def handle_qkv(sds_sd, ait_sd, sds_key, ait_keys, dims=None):
643643
old_state_dict,
644644
new_state_dict,
645645
old_key,
646-
[f"transformer.single_transformer_blocks.{block_num}.norm.linear"],
646+
[
647+
f"transformer.single_transformer_blocks.{block_num}.attn.to_q",
648+
f"transformer.single_transformer_blocks.{block_num}.attn.to_k",
649+
f"transformer.single_transformer_blocks.{block_num}.attn.to_v",
650+
],
647651
)
648652

649653
if "down" in old_key:

0 commit comments

Comments
 (0)