mirror of
https://github.com/invoke-ai/InvokeAI.git
synced 2026-04-23 03:00:31 -04:00
temp: remove omi dependency to allow publishing on PyPI
This commit is contained in:
@@ -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
|
||||
|
||||
@@ -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]
|
||||
|
||||
Reference in New Issue
Block a user