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

Add a app config parameter to control the ModelCache logging behavior.

Browse Source
This commit is contained in:
Ryan Dick
2023-11-02 12:07:48 -04:00
committed by Kent Keirsey
parent 3781e56e57
commit 4a683cc669
2 changed files with 6 additions and 4 deletions
Download Patch File Download Diff File Expand all files Collapse all files

7
invokeai/backend/model_management/model_manager.py
View File

@ -351,6 +351,7 @@ class ModelManager(object):
precision=precision,
sequential_offload=sequential_offload,
logger=logger,
log_memory_usage=self.app_config.log_memory_usage,
)
self._read_models(config)
@ -933,8 +934,7 @@ class ModelManager(object):
"""
Returns the preamble for the config file.
"""
return textwrap.dedent(
"""
return textwrap.dedent("""
# This file describes the alternative machine learning models
# available to InvokeAI script.
#
@ -942,8 +942,7 @@ class ModelManager(object):
# model requires a model config file, a weights file,
# and the width and height of the images it
# was trained on.
"""
)
""")
def scan_models_directory(
self,