From de5f413440378458df90dc16c55ce12ad879aa22 Mon Sep 17 00:00:00 2001 From: Billy Date: Tue, 24 Jun 2025 07:12:11 +1000 Subject: [PATCH] Filter bundle_emb for all LoRAs --- invokeai/backend/model_manager/load/model_loaders/lora.py | 4 ++++ 1 file changed, 4 insertions(+) diff --git a/invokeai/backend/model_manager/load/model_loaders/lora.py b/invokeai/backend/model_manager/load/model_loaders/lora.py index 13760dc11c..ca37492f15 100644 --- a/invokeai/backend/model_manager/load/model_loaders/lora.py +++ b/invokeai/backend/model_manager/load/model_loaders/lora.py @@ -80,6 +80,10 @@ class LoRALoader(ModelLoader): else: state_dict = torch.load(model_path, map_location="cpu") + # 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") } + # 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 [ BaseModelType.StableDiffusionXL,