Skip to content

Commit 363e35b

Browse files
temp: remove omi dependency to allow publishing on PyPI
1 parent 1c3be7e commit 363e35b

File tree

3 files changed

+1480
-1490
lines changed

3 files changed

+1480
-1490
lines changed
Lines changed: 11 additions & 10 deletions
Original file line numberDiff line numberDiff line change
@@ -1,16 +1,17 @@
1-
import omi_model_standards.convert.lora.convert_lora_util as lora_util
2-
from omi_model_standards.convert.lora.convert_flux_lora import convert_flux_lora_key_sets
3-
from omi_model_standards.convert.lora.convert_sdxl_lora import convert_sdxl_lora_key_sets
1+
# import omi_model_standards.convert.lora.convert_lora_util as lora_util
2+
# from omi_model_standards.convert.lora.convert_flux_lora import convert_flux_lora_key_sets
3+
# from omi_model_standards.convert.lora.convert_sdxl_lora import convert_sdxl_lora_key_sets
44

55
from invokeai.backend.model_manager.model_on_disk import StateDict
66
from invokeai.backend.model_manager.taxonomy import BaseModelType
77

88

99
def convert_from_omi(weights_sd: StateDict, base: BaseModelType):
10-
keyset = {
11-
BaseModelType.Flux: convert_flux_lora_key_sets(),
12-
BaseModelType.StableDiffusionXL: convert_sdxl_lora_key_sets(),
13-
}[base]
14-
source = "omi"
15-
target = "legacy_diffusers"
16-
return lora_util.__convert(weights_sd, keyset, source, target) # type: ignore
10+
raise NotImplementedError
11+
# keyset = {
12+
# BaseModelType.Flux: convert_flux_lora_key_sets(),
13+
# BaseModelType.StableDiffusionXL: convert_sdxl_lora_key_sets(),
14+
# }[base]
15+
# source = "omi"
16+
# target = "legacy_diffusers"
17+
# return lora_util.__convert(weights_sd, keyset, source, target) # type: ignore

pyproject.toml

Lines changed: 0 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -74,7 +74,6 @@ dependencies = [
7474
"python-multipart",
7575
"requests",
7676
"semver~=3.0.1",
77-
"omi-model-standards @ git+https://github.com/Open-Model-Initiative/OMI-Model-Standards.git@4ad235ceba6b42a97942834b7664379e4ec2d93c"
7877
]
7978

8079
[project.optional-dependencies]

0 commit comments

Comments
 (0)