From aaa34f717d9f424d328cb00a89a8f9dd82bc7ddd Mon Sep 17 00:00:00 2001
From: Billy <billy@invoke.ai>
Date: Thu, 26 Jun 2025 17:25:37 +1000
Subject: [PATCH] OMI files

---
 invokeai/backend/model_manager/omi/__init__.py |  6 ++++++
 invokeai/backend/model_manager/omi/omi.py      | 12 +++++++-----
 2 files changed, 13 insertions(+), 5 deletions(-)
 create mode 100644 invokeai/backend/model_manager/omi/__init__.py

diff --git a/invokeai/backend/model_manager/omi/__init__.py b/invokeai/backend/model_manager/omi/__init__.py
new file mode 100644
index 0000000000..7d04da561d
--- /dev/null
+++ b/invokeai/backend/model_manager/omi/__init__.py
@@ -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
diff --git a/invokeai/backend/model_manager/omi/omi.py b/invokeai/backend/model_manager/omi/omi.py
index e595e6d442..4a78e99b41 100644
--- a/invokeai/backend/model_manager/omi/omi.py
+++ b/invokeai/backend/model_manager/omi/omi.py
@@ -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"