mirror of
https://github.com/invoke-ai/InvokeAI
synced 2024-08-30 20:32:17 +00:00
0f8af643d1
We have two different classes named `ModelInfo` which might need to be used by API consumers. We need to export both but have to deal with this naming collision. The `ModelInfo` I've renamed here is the one that is returned when a model is loaded. It's the object least likely to be used by API consumers.
21 lines
549 B
Python
21 lines
549 B
Python
# 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
|