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

Convert FROM OMI to diffusers

Browse Source
This commit is contained in:
Billy
2025-06-19 11:00:27 +10:00
parent 4e98e7d0a2
commit 637487c573
2 changed files with 4 additions and 4 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

@ -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_from_omi(state_dict, config.base) # type: ignore
# Apply state_dict key conversions, if necessary.
if self._model_base == BaseModelType.StableDiffusionXL:
@ -101,7 +101,7 @@ class LoRALoader(ModelLoader):
elif is_state_dict_likely_flux_control(state_dict=state_dict):
model = lora_model_from_flux_control_state_dict(state_dict=state_dict)
else:
raise ValueError(f"LoRA model is in unsupported FLUX format: {config.format}")
raise ValueError(f"LoRA model is in unsupported FLUX format")
else:
raise ValueError(f"LoRA model is in unsupported FLUX format: {config.format}")
elif self._model_base in [BaseModelType.StableDiffusion1, BaseModelType.StableDiffusion2]:

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

@ -8,11 +8,11 @@ from invokeai.backend.model_manager.model_on_disk import StateDict
from invokeai.backend.model_manager.taxonomy import BaseModelType
def convert_to_omi(weights_sd: StateDict, base: 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]
return lora_util.convert_to_omi(weights_sd, keyset)
return lora_util.convert_to_diffusers(weights_sd, keyset)