wehub-resource-sync
cddb07a176
docs / deploy (push) Has been cancelled
docs / changes (push) Has been cancelled
docs / check-and-build (push) Has been cancelled
build container image / cpu (push) Has been cancelled
build container image / cuda (push) Has been cancelled
build container image / rocm (push) Has been cancelled
frontend checks / frontend-checks (push) Has been cancelled
frontend tests / frontend-tests (push) Has been cancelled
lfs checks / lfs-check (push) Has been cancelled
python checks / python-checks (push) Has been cancelled
python tests / py3.12: macos-default (push) Has been cancelled
python tests / py3.11: windows-cpu (push) Has been cancelled
python tests / py3.12: windows-cpu (push) Has been cancelled
python tests / py3.11: linux-cpu (push) Has been cancelled
typegen checks / typegen-checks (push) Has been cancelled
uv lock checks / uv-lock-checks (push) Has been cancelled
openapi checks / openapi-checks (push) Has been cancelled
python tests / py3.11: macos-default (push) Has been cancelled
python tests / py3.12: linux-cpu (push) Has been cancelled
28 lines
1015 B
Python
28 lines
1015 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 invokeai.backend.model_manager.load.load_base import LoadedModel, LoadedModelWithoutConfig, ModelLoaderBase
|
|
from invokeai.backend.model_manager.load.load_default import ModelLoader
|
|
from invokeai.backend.model_manager.load.model_cache.model_cache import ModelCache
|
|
from invokeai.backend.model_manager.load.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",
|
|
]
|