"...text-generation-inference.git" did not exist on "069895b9859b776cab8145f3fa6f6d16ac40af47"
Unverified Commit 2f7a417d authored by 赵三石's avatar 赵三石 Committed by GitHub
Browse files

Update lora_conversion_utils.py (#9980)



x-flux single-blocks lora load
Co-authored-by: default avatarSayak Paul <spsayakpaul@gmail.com>
Co-authored-by: default avatarYiYi Xu <yixu310@gmail.com>
parent 4450d26b
...@@ -643,7 +643,11 @@ def _convert_xlabs_flux_lora_to_diffusers(old_state_dict): ...@@ -643,7 +643,11 @@ def _convert_xlabs_flux_lora_to_diffusers(old_state_dict):
old_state_dict, old_state_dict,
new_state_dict, new_state_dict,
old_key, old_key,
[f"transformer.single_transformer_blocks.{block_num}.norm.linear"], [
f"transformer.single_transformer_blocks.{block_num}.attn.to_q",
f"transformer.single_transformer_blocks.{block_num}.attn.to_k",
f"transformer.single_transformer_blocks.{block_num}.attn.to_v",
],
) )
if "down" in old_key: if "down" in old_key:
......
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