From 2832ca300fb05d29d0de72b2896607c26800fa87 Mon Sep 17 00:00:00 2001 From: Billy Date: Tue, 24 Jun 2025 07:26:42 +1000 Subject: [PATCH] Formatting --- invokeai/backend/model_manager/load/model_loaders/lora.py | 2 +- .../patches/lora_conversions/sdxl_lora_conversion_utils.py | 1 - 2 files changed, 1 insertion(+), 2 deletions(-) diff --git a/invokeai/backend/model_manager/load/model_loaders/lora.py b/invokeai/backend/model_manager/load/model_loaders/lora.py index ca37492f15..275efe8a22 100644 --- a/invokeai/backend/model_manager/load/model_loaders/lora.py +++ b/invokeai/backend/model_manager/load/model_loaders/lora.py @@ -82,7 +82,7 @@ class LoRALoader(ModelLoader): # Strip 'bundle_emb' keys - these are unused and currently cause downstream errors. # To revisit later to determine if they're needed/useful. - state_dict = { k: v for k, v in state_dict.items() if not k.startswith("bundle_emb") } + state_dict = {k: v for k, v in state_dict.items() if not k.startswith("bundle_emb")} # At the time of writing, we support the OMI standard for base models Flux and SDXL if config.format == ModelFormat.OMI and self._model_base in [ diff --git a/invokeai/backend/patches/lora_conversions/sdxl_lora_conversion_utils.py b/invokeai/backend/patches/lora_conversions/sdxl_lora_conversion_utils.py index bee9fb8d5e..26aa177215 100644 --- a/invokeai/backend/patches/lora_conversions/sdxl_lora_conversion_utils.py +++ b/invokeai/backend/patches/lora_conversions/sdxl_lora_conversion_utils.py @@ -35,7 +35,6 @@ 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(): - 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.