Mirrored_Repos/InvokeAI
1
0
Fork 0
mirror of https://github.com/invoke-ai/InvokeAI synced 2024-08-30 20:32:17 +00:00
Code Issues Packages Projects Releases Wiki Activity

all features implemented, docs updated, ready for review

Browse Source
This commit is contained in:
Lincoln Stein
2023-11-26 13:18:21 -05:00
parent dc5c452ef9
commit 8695ad6f59
12 changed files with 542 additions and 469 deletions
Download Patch File Download Diff File Expand all files Collapse all files

12
invokeai/backend/model_manager/__init__.py
View File

@ -1,17 +1,17 @@
"""Re-export frequently-used symbols from the Model Manager backend."""
from .probe import ModelProbe
from .config import (
AnyModelConfig,
BaseModelType,
InvalidModelConfigException,
ModelConfigFactory,
BaseModelType,
ModelType,
SubModelType,
ModelVariantType,
ModelFormat,
ModelType,
ModelVariantType,
SchedulerPredictionType,
AnyModelConfig,
SubModelType,
)
from .probe import ModelProbe
from .search import ModelSearch
__all__ = ['ModelProbe', 'ModelSearch',

2
invokeai/backend/util/__init__.py
View File

@ -11,7 +11,7 @@ from .devices import ( # noqa: F401
normalize_device,
torch_dtype,
)
from .util import Chdir, ask_user, download_with_resume, instantiate_from_config, url_attachment_name # noqa: F401
from .logging import InvokeAILogger
from .util import Chdir, ask_user, download_with_resume, instantiate_from_config, url_attachment_name # noqa: F401
__all__ = ['Chdir', 'InvokeAILogger', 'choose_precision', 'choose_torch_device']