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

Formatting

Browse Source
This commit is contained in:
Billy
2025-06-19 09:40:58 +10:00
parent 45d09f8f51
commit 12f65d800d
2 changed files with 8 additions and 7 deletions
Download Patch File Download Diff File Expand all files Collapse all files

4
invokeai/backend/model_manager/load/model_loaders/lora.py
View File

@ -39,11 +39,11 @@ from invokeai.backend.patches.lora_conversions.flux_onetrainer_lora_conversion_u
from invokeai.backend.patches.lora_conversions.sd_lora_conversion_utils import lora_model_from_sd_state_dict
from invokeai.backend.patches.lora_conversions.sdxl_lora_conversion_utils import convert_sdxl_keys_to_diffusers_format
@ModelLoaderRegistry.register(base=BaseModelType.Flux, type=ModelType.LoRA, format=ModelFormat.OMI)
@ModelLoaderRegistry.register(base=BaseModelType.StableDiffusion1, type=ModelType.LoRA, format=ModelFormat.OMI)
@ModelLoaderRegistry.register(base=BaseModelType.StableDiffusion3, type=ModelType.LoRA, format=ModelFormat.OMI)
@ModelLoaderRegistry.register(base=BaseModelType.StableDiffusionXL, type=ModelType.LoRA, format=ModelFormat.OMI)
@ModelLoaderRegistry.register(base=BaseModelType.Any, type=ModelType.LoRA, format=ModelFormat.Diffusers)
@ModelLoaderRegistry.register(base=BaseModelType.Any, type=ModelType.LoRA, format=ModelFormat.LyCORIS)
@ModelLoaderRegistry.register(base=BaseModelType.Flux, type=ModelType.ControlLoRa, format=ModelFormat.LyCORIS)
@ -79,7 +79,7 @@ class LoRALoader(ModelLoader):
state_dict = torch.load(model_path, map_location="cpu")
if config.format == ModelFormat.OMI:
state_dict = convert_to_omi(state_dict. config.base) # type: ignore
state_dict = convert_to_omi(state_dict.config.base) # type: ignore
# Apply state_dict key conversions, if necessary.
if self._model_base == BaseModelType.StableDiffusionXL:

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

@ -1,10 +1,11 @@
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
from omi_model_standards.convert.lora.convert_sdxl_lora import convert_sdxl_lora_key_sets
from omi_model_standards.convert.lora.convert_flux_lora import convert_flux_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_sd3_lora import convert_sd3_lora_key_sets
import omi_model_standards.convert.lora.convert_lora_util as lora_util
def convert_to_omi(weights_sd: StateDict, base: BaseModelType):