Mirrored_Repos/InvokeAI
1
0
Fork 0
mirror of https://github.com/invoke-ai/InvokeAI synced 2025-07-26 05:17:55 +00:00
Code Issues Packages Projects Releases Wiki Activity

OMI files

Browse Source
This commit is contained in:
Billy
2025-06-26 17:25:37 +10:00
committed by psychedelicious
parent fe83c2f81f
commit aaa34f717d
2 changed files with 13 additions and 5 deletions
Download Patch File Download Diff File Expand all files Collapse all files

6
invokeai/backend/model_manager/omi/__init__.py Normal file
View File

@ -0,0 +1,6 @@
from invokeai.backend.model_manager.omi.vendor.model_spec.architecture import (
stable_diffusion_xl_1_lora,
flux_dev_1_lora
)
from invokeai.backend.model_manager.omi.omi import convert_from_omi

12
invokeai/backend/model_manager/omi/omi.py
View File

@ -1,6 +1,8 @@
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_sdxl_lora import convert_sdxl_lora_key_sets
from invokeai.backend.model_manager.omi.vendor.convert.lora import (
convert_lora_util as lora_util,
convert_flux_lora as omi_flux,
convert_sdxl_lora as omi_sdxl,
)
from invokeai.backend.model_manager.model_on_disk import StateDict
from invokeai.backend.model_manager.taxonomy import BaseModelType
@ -8,8 +10,8 @@ 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.Flux: omi_flux.convert_flux_lora_key_sets(),
BaseModelType.StableDiffusionXL: omi_sdxl.convert_sdxl_lora_key_sets(),
}[base]
source = "omi"
target = "legacy_diffusers"