InvokeAI/invokeai/backend/model_management/__init__.py

# ruff: noqa: I001, F401
"""
Initialization file for invokeai.backend.model_management
"""
# This import must be first
from .model_manager import AddModelResult, LoadedModelInfo, ModelManager, SchedulerPredictionType
from .lora import ModelPatcher, ONNXModelPatcher
from .model_cache import ModelCache

from .models import (
    BaseModelType,
    DuplicateModelException,
    ModelNotFoundException,
    ModelType,
    ModelVariantType,
    SubModelType,
)

# This import must be last
from .model_merge import MergeInterpolationMethod, ModelMerger