mirror of
https://github.com/invoke-ai/InvokeAI.git
synced 2026-04-23 03:00:31 -04:00
5d612ec095
- Rename old "model_management" directory to "model_management_OLD" in order to catch dangling references to original model manager. - Caught and fixed most dangling references (still checking) - Rename lora, textual_inversion and model_patcher modules - Introduce a RawModel base class to simplfy the Union returned by the model loaders. - Tidy up the model manager 2-related tests. Add useful fixtures, and a finalizer to the queue and installer fixtures that will stop the services and release threads.
18 lines
491 B
Python
18 lines
491 B
Python
"""Initialization file for model manager service."""
|
|
|
|
from invokeai.backend.model_manager import AnyModel, AnyModelConfig, BaseModelType, ModelType, SubModelType
|
|
from invokeai.backend.model_manager.load import LoadedModel
|
|
|
|
from .model_manager_default import ModelManagerServiceBase, ModelManagerService
|
|
|
|
__all__ = [
|
|
"ModelManagerServiceBase",
|
|
"ModelManagerService",
|
|
"AnyModel",
|
|
"AnyModelConfig",
|
|
"BaseModelType",
|
|
"ModelType",
|
|
"SubModelType",
|
|
"LoadedModel",
|
|
]
|