Skip to content

Commit 3159e60

Browse files
fix xlabs FLUX lora conversion typo (#9581)
* fix startswith syntax in xlabs lora conversion * Trigger CI #9581 (comment)
1 parent 99f6082 commit 3159e60

File tree

1 file changed

+1
-1
lines changed

1 file changed

+1
-1
lines changed

src/diffusers/loaders/lora_conversion_utils.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -632,7 +632,7 @@ def handle_qkv(sds_sd, ait_sd, sds_key, ait_keys, dims=None):
632632
new_key += ".lora_B.weight"
633633

634634
# Handle single_blocks
635-
elif old_key.startswith("diffusion_model.single_blocks", "single_blocks"):
635+
elif old_key.startswith(("diffusion_model.single_blocks", "single_blocks")):
636636
block_num = re.search(r"single_blocks\.(\d+)", old_key).group(1)
637637
new_key = f"transformer.single_transformer_blocks.{block_num}"
638638

0 commit comments

Comments
 (0)