mirror of
https://github.com/invoke-ai/InvokeAI.git
synced 2026-02-11 20:05:13 -05:00
996eb96b4e
- ModelMetadataStoreService is now injected into ModelRecordStoreService (these two services are really joined at the hip, and should someday be merged) - ModelRecordStoreService is now injected into ModelManagerService - Reduced timeout value for the various installer and download wait*() methods - Introduced a Mock modelmanager for testing - Removed bare print() statement with _logger in the install helper backend. - Removed unused code from model loader init file - Made `locker` a private variable in the `LoadedModel` object. - Fixed up model merge frontend (will be deprecated anyway!)
66 lines
2.3 KiB
Python
66 lines
2.3 KiB
Python
# Copyright (c) 2023 Lincoln D. Stein and the InvokeAI Development Team
|
|
"""
|
|
Storage for Model Metadata
|
|
"""
|
|
|
|
from abc import ABC, abstractmethod
|
|
from typing import List, Set, Tuple
|
|
|
|
from invokeai.backend.model_manager.metadata import AnyModelRepoMetadata
|
|
|
|
|
|
class ModelMetadataStoreBase(ABC):
|
|
"""Store, search and fetch model metadata retrieved from remote repositories."""
|
|
|
|
@abstractmethod
|
|
def add_metadata(self, model_key: str, metadata: AnyModelRepoMetadata) -> None:
|
|
"""
|
|
Add a block of repo metadata to a model record.
|
|
|
|
The model record config must already exist in the database with the
|
|
same key. Otherwise a FOREIGN KEY constraint exception will be raised.
|
|
|
|
:param model_key: Existing model key in the `model_config` table
|
|
:param metadata: ModelRepoMetadata object to store
|
|
"""
|
|
|
|
@abstractmethod
|
|
def get_metadata(self, model_key: str) -> AnyModelRepoMetadata:
|
|
"""Retrieve the ModelRepoMetadata corresponding to model key."""
|
|
|
|
@abstractmethod
|
|
def list_all_metadata(self) -> List[Tuple[str, AnyModelRepoMetadata]]: # key, metadata
|
|
"""Dump out all the metadata."""
|
|
|
|
@abstractmethod
|
|
def update_metadata(self, model_key: str, metadata: AnyModelRepoMetadata) -> AnyModelRepoMetadata:
|
|
"""
|
|
Update metadata corresponding to the model with the indicated key.
|
|
|
|
:param model_key: Existing model key in the `model_config` table
|
|
:param metadata: ModelRepoMetadata object to update
|
|
"""
|
|
|
|
@abstractmethod
|
|
def list_tags(self) -> Set[str]:
|
|
"""Return all tags in the tags table."""
|
|
|
|
@abstractmethod
|
|
def search_by_tag(self, tags: Set[str]) -> Set[str]:
|
|
"""Return the keys of models containing all of the listed tags."""
|
|
|
|
@abstractmethod
|
|
def search_by_author(self, author: str) -> Set[str]:
|
|
"""Return the keys of models authored by the indicated author."""
|
|
|
|
@abstractmethod
|
|
def search_by_name(self, name: str) -> Set[str]:
|
|
"""
|
|
Return the keys of models with the indicated name.
|
|
|
|
Note that this is the name of the model given to it by
|
|
the remote source. The user may have changed the local
|
|
name. The local name will be located in the model config
|
|
record object.
|
|
"""
|