mirror of
https://github.com/invoke-ai/InvokeAI.git
synced 2026-01-21 05:58:05 -05:00
a4cddfa47d
Adds full support for managing model-to-model relationships in the UI and backend.
Introduces RelatedModels subpanel for linking and unlinking models in model management.
- Adds REST API routes for adding, removing, and retrieving model relationships.
- New database migration: creates model_relationships table for bidirectional links.
- New service layer (model_relationships) for relationship management.
- Updated frontend: Related models float to top of LoRA/Main grouped model comboboxes for quick access.
- Added 'Show Only Related' toggle badge to MainModelPicker filter bar
**Amended commit to remove changes to ParamMainModelSelect.tsx and MainModelPicker.tsx to avoid conflict with upstream deletion/ rewrite**
69 lines
4.2 KiB
Python
69 lines
4.2 KiB
Python
from logging import Logger
|
|
|
|
from invokeai.app.services.config.config_default import InvokeAIAppConfig
|
|
from invokeai.app.services.image_files.image_files_base import ImageFileStorageBase
|
|
from invokeai.app.services.shared.sqlite.sqlite_database import SqliteDatabase
|
|
from invokeai.app.services.shared.sqlite_migrator.migrations.migration_1 import build_migration_1
|
|
from invokeai.app.services.shared.sqlite_migrator.migrations.migration_2 import build_migration_2
|
|
from invokeai.app.services.shared.sqlite_migrator.migrations.migration_3 import build_migration_3
|
|
from invokeai.app.services.shared.sqlite_migrator.migrations.migration_4 import build_migration_4
|
|
from invokeai.app.services.shared.sqlite_migrator.migrations.migration_5 import build_migration_5
|
|
from invokeai.app.services.shared.sqlite_migrator.migrations.migration_6 import build_migration_6
|
|
from invokeai.app.services.shared.sqlite_migrator.migrations.migration_7 import build_migration_7
|
|
from invokeai.app.services.shared.sqlite_migrator.migrations.migration_8 import build_migration_8
|
|
from invokeai.app.services.shared.sqlite_migrator.migrations.migration_9 import build_migration_9
|
|
from invokeai.app.services.shared.sqlite_migrator.migrations.migration_10 import build_migration_10
|
|
from invokeai.app.services.shared.sqlite_migrator.migrations.migration_11 import build_migration_11
|
|
from invokeai.app.services.shared.sqlite_migrator.migrations.migration_12 import build_migration_12
|
|
from invokeai.app.services.shared.sqlite_migrator.migrations.migration_13 import build_migration_13
|
|
from invokeai.app.services.shared.sqlite_migrator.migrations.migration_14 import build_migration_14
|
|
from invokeai.app.services.shared.sqlite_migrator.migrations.migration_15 import build_migration_15
|
|
from invokeai.app.services.shared.sqlite_migrator.migrations.migration_16 import build_migration_16
|
|
from invokeai.app.services.shared.sqlite_migrator.migrations.migration_17 import build_migration_17
|
|
from invokeai.app.services.shared.sqlite_migrator.migrations.migration_18 import build_migration_18
|
|
from invokeai.app.services.shared.sqlite_migrator.migrations.migration_19 import build_migration_19
|
|
from invokeai.app.services.shared.sqlite_migrator.migrations.migration_20 import build_migration_20
|
|
from invokeai.app.services.shared.sqlite_migrator.sqlite_migrator_impl import SqliteMigrator
|
|
|
|
|
|
def init_db(config: InvokeAIAppConfig, logger: Logger, image_files: ImageFileStorageBase) -> SqliteDatabase:
|
|
"""
|
|
Initializes the SQLite database.
|
|
|
|
:param config: The app config
|
|
:param logger: The logger
|
|
:param image_files: The image files service (used by migration 2)
|
|
|
|
This function:
|
|
- Instantiates a :class:`SqliteDatabase`
|
|
- Instantiates a :class:`SqliteMigrator` and registers all migrations
|
|
- Runs all migrations
|
|
"""
|
|
db_path = None if config.use_memory_db else config.db_path
|
|
db = SqliteDatabase(db_path=db_path, logger=logger, verbose=config.log_sql)
|
|
|
|
migrator = SqliteMigrator(db=db)
|
|
migrator.register_migration(build_migration_1())
|
|
migrator.register_migration(build_migration_2(image_files=image_files, logger=logger))
|
|
migrator.register_migration(build_migration_3(app_config=config, logger=logger))
|
|
migrator.register_migration(build_migration_4())
|
|
migrator.register_migration(build_migration_5())
|
|
migrator.register_migration(build_migration_6())
|
|
migrator.register_migration(build_migration_7())
|
|
migrator.register_migration(build_migration_8(app_config=config))
|
|
migrator.register_migration(build_migration_9())
|
|
migrator.register_migration(build_migration_10())
|
|
migrator.register_migration(build_migration_11(app_config=config, logger=logger))
|
|
migrator.register_migration(build_migration_12(app_config=config))
|
|
migrator.register_migration(build_migration_13())
|
|
migrator.register_migration(build_migration_14())
|
|
migrator.register_migration(build_migration_15())
|
|
migrator.register_migration(build_migration_16())
|
|
migrator.register_migration(build_migration_17())
|
|
migrator.register_migration(build_migration_18())
|
|
migrator.register_migration(build_migration_19(app_config=config))
|
|
migrator.register_migration(build_migration_20())
|
|
migrator.run_migrations()
|
|
|
|
return db
|