InvokeAI/invokeai/backend/model_management/__init__.py

"""
Initialization file for invokeai.backend.model_management
"""
# This import must be first
from .model_manager import ModelManager, ModelInfo, AddModelResult, SchedulerPredictionType  # noqa: F401 isort: split

from .lora import ModelPatcher, ONNXModelPatcher  # noqa: F401
from .model_cache import ModelCache  # noqa: F401
from .models import (  # noqa: F401
    BaseModelType,
    DuplicateModelException,
    ModelNotFoundException,
    ModelType,
    ModelVariantType,
    SubModelType,
)

# This import must be last
from .model_merge import ModelMerger, MergeInterpolationMethod  # noqa: F401 isort: split