mirror of
https://github.com/invoke-ai/InvokeAI.git
synced 2026-02-02 07:25:07 -05:00
3e0fb45dd7
* use model_class.load_singlefile() instead of converting; works, but performance is poor * adjust the convert api - not right just yet * working, needs sql migrator update * rename migration_11 before conflict merge with main * Update invokeai/backend/model_manager/load/model_loaders/stable_diffusion.py Co-authored-by: Ryan Dick <ryanjdick3@gmail.com> * Update invokeai/backend/model_manager/load/model_loaders/stable_diffusion.py Co-authored-by: Ryan Dick <ryanjdick3@gmail.com> * implement lightweight version-by-version config migration * simplified config schema migration code * associate sdxl config with sdxl VAEs * remove use of original_config_file in load_single_file() --------- Co-authored-by: Lincoln Stein <lstein@gmail.com> Co-authored-by: Ryan Dick <ryanjdick3@gmail.com>
28 lines
883 B
Python
28 lines
883 B
Python
# Copyright (c) 2024 Lincoln D. Stein and the InvokeAI Development Team
|
|
"""
|
|
Init file for the model loader.
|
|
"""
|
|
|
|
from importlib import import_module
|
|
from pathlib import Path
|
|
|
|
from .load_base import LoadedModel, LoadedModelWithoutConfig, ModelLoaderBase
|
|
from .load_default import ModelLoader
|
|
from .model_cache.model_cache_default import ModelCache
|
|
from .model_loader_registry import ModelLoaderRegistry, ModelLoaderRegistryBase
|
|
|
|
# This registers the subclasses that implement loaders of specific model types
|
|
loaders = [x.stem for x in Path(Path(__file__).parent, "model_loaders").glob("*.py") if x.stem != "__init__"]
|
|
for module in loaders:
|
|
import_module(f"{__package__}.model_loaders.{module}")
|
|
|
|
__all__ = [
|
|
"LoadedModel",
|
|
"LoadedModelWithoutConfig",
|
|
"ModelCache",
|
|
"ModelLoaderBase",
|
|
"ModelLoader",
|
|
"ModelLoaderRegistryBase",
|
|
"ModelLoaderRegistry",
|
|
]
|