Mirrored_Repos/InvokeAI
1
0
Fork 0
mirror of https://github.com/invoke-ai/InvokeAI synced 2025-07-26 05:17:55 +00:00
Code Issues Packages Projects Releases Wiki Activity

Remove bundle_emb filter

Browse Source
This commit is contained in:
Billy
2025-06-24 06:53:33 +10:00
parent 77e029a49f
commit fbc14c61ea
1 changed files with 2 additions and 5 deletions
Download Patch File Download Diff File Expand all files Collapse all files

7
invokeai/backend/patches/lora_conversions/sdxl_lora_conversion_utils.py
View File

@ -22,7 +22,7 @@ def convert_sdxl_keys_to_diffusers_format(state_dict: Dict[str, T]) -> dict[str,
ValueError: If state_dict contains an unrecognized key, or not all keys could be converted.
Returns:
Dict[str, Tensor]: The diffusers-format state_dict with bundle_emb keys removed.
Dict[str, Tensor]: The diffusers-format state_dict.
"""
converted_count = 0 # The number of Stability AI keys converted to diffusers format.
not_converted_count = 0 # The number of keys that were not converted.
@ -35,10 +35,7 @@ def convert_sdxl_keys_to_diffusers_format(state_dict: Dict[str, T]) -> dict[str,
new_state_dict: dict[str, T] = {}
for full_key, value in state_dict.items():
# Skip keys that start with "bundle_emb"
if full_key.startswith("bundle_emb"):
continue
if full_key.startswith("lora_unet_"):
search_key = full_key.replace("lora_unet_", "")
# Use bisect to find the key in stability_unet_keys that *may* match the search_key's prefix.