diff --git a/invokeai/backend/model_manager/omi.py b/invokeai/backend/model_manager/omi.py index a33571a415..ff62ad52ec 100644 --- a/invokeai/backend/model_manager/omi.py +++ b/invokeai/backend/model_manager/omi.py @@ -15,4 +15,4 @@ def convert_from_omi(weights_sd: StateDict, base: BaseModelType): BaseModelType.StableDiffusion1: convert_sd_lora_key_sets(), BaseModelType.StableDiffusion3: convert_sd3_lora_key_sets(), }[base] - return lora_util.convert_to_diffusers(weights_sd, keyset) + return lora_util.convert_to_diffusers(weights_sd, keyset) # type: ignore