diff --git a/invokeai/backend/install/install_helper.py b/invokeai/backend/install/install_helper.py index 975088409c..f3fc82301e 100644 --- a/invokeai/backend/install/install_helper.py +++ b/invokeai/backend/install/install_helper.py @@ -19,7 +19,6 @@ from invokeai.app.services.model_install import ( ModelInstallService, ModelInstallServiceBase, ) -from invokeai.app.services.model_metadata import ModelMetadataStoreSQL from invokeai.app.services.model_records import ModelRecordServiceBase, ModelRecordServiceSQL from invokeai.app.services.shared.sqlite.sqlite_util import init_db from invokeai.backend.model_manager import ( @@ -39,7 +38,7 @@ def initialize_record_store(app_config: InvokeAIAppConfig) -> ModelRecordService logger = InvokeAILogger.get_logger(config=app_config) image_files = DiskImageFileStorage(f"{app_config.output_path}/images") db = init_db(config=app_config, logger=logger, image_files=image_files) - obj: ModelRecordServiceBase = ModelRecordServiceSQL(db, ModelMetadataStoreSQL(db)) + obj: ModelRecordServiceBase = ModelRecordServiceSQL(db) return obj diff --git a/invokeai/frontend/merge/merge_diffusers.py b/invokeai/frontend/merge/merge_diffusers.py index 78b5df859b..5cc72fde74 100644 --- a/invokeai/frontend/merge/merge_diffusers.py +++ b/invokeai/frontend/merge/merge_diffusers.py @@ -20,7 +20,6 @@ from invokeai.app.services.config import InvokeAIAppConfig from invokeai.app.services.download import DownloadQueueService from invokeai.app.services.image_files.image_files_disk import DiskImageFileStorage from invokeai.app.services.model_install import ModelInstallService -from invokeai.app.services.model_metadata import ModelMetadataStoreSQL from invokeai.app.services.model_records import ModelRecordServiceBase, ModelRecordServiceSQL from invokeai.app.services.shared.sqlite.sqlite_util import init_db from invokeai.backend.model_manager import ( @@ -413,7 +412,7 @@ def get_config_store() -> ModelRecordServiceSQL: assert output_path is not None image_files = DiskImageFileStorage(output_path / "images") db = init_db(config=config, logger=InvokeAILogger.get_logger(), image_files=image_files) - return ModelRecordServiceSQL(db, ModelMetadataStoreSQL(db)) + return ModelRecordServiceSQL(db) def get_model_merger(record_store: ModelRecordServiceBase) -> ModelMerger: