Formatting

This commit is contained in:
Billy
2025-06-23 13:52:19 +10:00
parent 62c3b01e4f
commit 150a876c73
2 changed files with 5 additions and 4 deletions

View File

@ -81,7 +81,10 @@ class LoRALoader(ModelLoader):
state_dict = torch.load(model_path, map_location="cpu")
# At the time of writing, we support the OMI standard for base models Flux and SDXL
if config.format == ModelFormat.OMI and self._model_base in [BaseModelType.StableDiffusionXL, BaseModelType.Flux]:
if config.format == ModelFormat.OMI and self._model_base in [
BaseModelType.StableDiffusionXL,
BaseModelType.Flux,
]:
state_dict = convert_from_omi(state_dict, config.base) # type: ignore
# Apply state_dict key conversions, if necessary.

View File

@ -1,7 +1,5 @@
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
@ -15,4 +13,4 @@ def convert_from_omi(weights_sd: StateDict, base: BaseModelType):
}[base]
source = "omi"
target = "legacy_diffusers"
return lora_util.__convert(weights_sd, keyset, source, target) # type: ignore
return lora_util.__convert(weights_sd, keyset, source, target) # type: ignore