wehub-resource-sync
cddb07a176
docs / deploy (push) Has been cancelled
docs / changes (push) Has been cancelled
docs / check-and-build (push) Has been cancelled
build container image / cpu (push) Has been cancelled
build container image / cuda (push) Has been cancelled
build container image / rocm (push) Has been cancelled
frontend checks / frontend-checks (push) Has been cancelled
frontend tests / frontend-tests (push) Has been cancelled
lfs checks / lfs-check (push) Has been cancelled
python checks / python-checks (push) Has been cancelled
python tests / py3.12: macos-default (push) Has been cancelled
python tests / py3.11: windows-cpu (push) Has been cancelled
python tests / py3.12: windows-cpu (push) Has been cancelled
python tests / py3.11: linux-cpu (push) Has been cancelled
typegen checks / typegen-checks (push) Has been cancelled
uv lock checks / uv-lock-checks (push) Has been cancelled
openapi checks / openapi-checks (push) Has been cancelled
python tests / py3.11: macos-default (push) Has been cancelled
python tests / py3.12: linux-cpu (push) Has been cancelled
32 lines
1.6 KiB
Python
32 lines
1.6 KiB
Python
from invokeai.app.services.invoker import Invoker
|
|
from invokeai.app.services.model_relationships.model_relationships_base import ModelRelationshipsServiceABC
|
|
from invokeai.backend.model_manager.configs.factory import AnyModelConfig
|
|
|
|
|
|
class ModelRelationshipsService(ModelRelationshipsServiceABC):
|
|
__invoker: Invoker
|
|
|
|
def start(self, invoker: Invoker) -> None:
|
|
self.__invoker = invoker
|
|
|
|
def add_model_relationship(self, model_key_1: str, model_key_2: str) -> None:
|
|
self.__invoker.services.model_relationship_records.add_model_relationship(model_key_1, model_key_2)
|
|
|
|
def remove_model_relationship(self, model_key_1: str, model_key_2: str) -> None:
|
|
self.__invoker.services.model_relationship_records.remove_model_relationship(model_key_1, model_key_2)
|
|
|
|
def get_related_model_keys(self, model_key: str) -> list[str]:
|
|
return self.__invoker.services.model_relationship_records.get_related_model_keys(model_key)
|
|
|
|
def add_relationship_from_models(self, model_1: AnyModelConfig, model_2: AnyModelConfig) -> None:
|
|
self.add_model_relationship(model_1.key, model_2.key)
|
|
|
|
def remove_relationship_from_models(self, model_1: AnyModelConfig, model_2: AnyModelConfig) -> None:
|
|
self.remove_model_relationship(model_1.key, model_2.key)
|
|
|
|
def get_related_keys_from_model(self, model: AnyModelConfig) -> list[str]:
|
|
return self.get_related_model_keys(model.key)
|
|
|
|
def get_related_model_keys_batch(self, model_keys: list[str]) -> list[str]:
|
|
return self.__invoker.services.model_relationship_records.get_related_model_keys_batch(model_keys)
|