temp: remove omi dependency to allow publishing on PyPI

This commit is contained in:
psychedelicious
2025-06-25 17:17:45 +10:00
parent 1c3be7e5b4
commit 363e35b6a7
3 changed files with 1453 additions and 1463 deletions

View File

@@ -1,16 +1,17 @@
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
# 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.model_on_disk import StateDict
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(),
}[base]
source = "omi"
target = "legacy_diffusers"
return lora_util.__convert(weights_sd, keyset, source, target) # type: ignore
raise NotImplementedError
# keyset = {
# BaseModelType.Flux: convert_flux_lora_key_sets(),
# BaseModelType.StableDiffusionXL: convert_sdxl_lora_key_sets(),
# }[base]
# source = "omi"
# target = "legacy_diffusers"
# return lora_util.__convert(weights_sd, keyset, source, target) # type: ignore

View File

@@ -74,7 +74,6 @@ dependencies = [
"python-multipart",
"requests",
"semver~=3.0.1",
"omi-model-standards @ git+https://github.com/Open-Model-Initiative/OMI-Model-Standards.git@4ad235ceba6b42a97942834b7664379e4ec2d93c"
]
[project.optional-dependencies]

2894
uv.lock generated

File diff suppressed because it is too large Load Diff