mirror of
https://github.com/invoke-ai/InvokeAI.git
synced 2026-02-13 07:45:02 -05:00
21 lines
1.0 KiB
Python
21 lines
1.0 KiB
Python
import omi_model_standards.convert.lora.convert_lora_util as lora_util
|
|
from omi_model_standards.convert.lora.convert_flux_lora import convert_flux_lora_key_sets
|
|
from omi_model_standards.convert.lora.convert_sd3_lora import convert_sd3_lora_key_sets
|
|
from omi_model_standards.convert.lora.convert_sd_lora import convert_sd_lora_key_sets
|
|
from omi_model_standards.convert.lora.convert_sdxl_lora import convert_sdxl_lora_key_sets
|
|
|
|
from invokeai.backend.model_manager.model_on_disk import StateDict
|
|
from invokeai.backend.model_manager.taxonomy import BaseModelType
|
|
|
|
|
|
def convert_from_omi(weights_sd: StateDict, base: BaseModelType):
|
|
keyset = {
|
|
BaseModelType.Flux: convert_flux_lora_key_sets(),
|
|
BaseModelType.StableDiffusionXL: convert_sdxl_lora_key_sets(),
|
|
BaseModelType.StableDiffusion1: convert_sd_lora_key_sets(),
|
|
BaseModelType.StableDiffusion3: convert_sd3_lora_key_sets(),
|
|
}[base]
|
|
|
|
target = "diffusers" # alternatively, "legacy_diffusers"
|
|
return lora_util.__convert(weights_sd, keyset, "omi", target) # type: ignore
|