From 58a681ade97457c71ab089653ef0eecf83141e63 Mon Sep 17 00:00:00 2001 From: =?UTF-8?q?Cl=C3=A9ment=20Leli=C3=A8vre?= Date: Fri, 4 Oct 2024 12:48:58 +0000 Subject: [PATCH 1/2] fix startswith syntax in xlabs lora conversion --- src/diffusers/loaders/lora_conversion_utils.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/src/diffusers/loaders/lora_conversion_utils.py b/src/diffusers/loaders/lora_conversion_utils.py index d829cc3a844b..d0ca40213b14 100644 --- a/src/diffusers/loaders/lora_conversion_utils.py +++ b/src/diffusers/loaders/lora_conversion_utils.py @@ -632,7 +632,7 @@ def handle_qkv(sds_sd, ait_sd, sds_key, ait_keys, dims=None): new_key += ".lora_B.weight" # Handle single_blocks - elif old_key.startswith("diffusion_model.single_blocks", "single_blocks"): + elif old_key.startswith(("diffusion_model.single_blocks", "single_blocks")): block_num = re.search(r"single_blocks\.(\d+)", old_key).group(1) new_key = f"transformer.single_transformer_blocks.{block_num}" From fac6547da0b9b57553b378bbb92801e6ecf863dc Mon Sep 17 00:00:00 2001 From: =?UTF-8?q?Cl=C3=A9ment=20Leli=C3=A8vre?= Date: Sun, 6 Oct 2024 19:04:31 +0000 Subject: [PATCH 2/2] Trigger CI https://github.com/huggingface/diffusers/pull/9581#issuecomment-2395530360