diff --git a/invokeai/backend/model_manager/omi.py b/invokeai/backend/model_manager/omi.py index ff62ad52ec..db12bcddc9 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) # type: ignore + return lora_util.__convert(weights_sd, keyset, "omi", "diffusers")