Unverified Commit 3159e60d authored by Clem's avatar Clem Committed by GitHub
Browse files

fix xlabs FLUX lora conversion typo (#9581)

* fix startswith syntax in xlabs lora conversion

* Trigger CI

https://github.com/huggingface/diffusers/pull/9581#issuecomment-2395530360
parent 99f60821
...@@ -632,7 +632,7 @@ def _convert_xlabs_flux_lora_to_diffusers(old_state_dict): ...@@ -632,7 +632,7 @@ def _convert_xlabs_flux_lora_to_diffusers(old_state_dict):
new_key += ".lora_B.weight" new_key += ".lora_B.weight"
# Handle single_blocks # 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) block_num = re.search(r"single_blocks\.(\d+)", old_key).group(1)
new_key = f"transformer.single_transformer_blocks.{block_num}" new_key = f"transformer.single_transformer_blocks.{block_num}"
......
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment