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

create databases directory on startup (#3518)

Browse Source 
This PR creates the databases directory at app startup time. It also
removes a couple of debugging statements that were inadvertently left in
the model manager.
This commit is contained in:
blessedcoolant 2023-06-08 23:40:32 +12:00 committed by GitHub
commit fe8f3381fc
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23
3 changed files with 3 additions and 7 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

1
invokeai/app/api/dependencies.py
View File

@ -57,6 +57,7 @@ class ApiDependencies:
# TODO: build a file/path manager? # TODO: build a file/path manager?
db_location = config.db_path db_location = config.db_path
db_location.parent.mkdir(parents=True,exist_ok=True)
graph_execution_manager = SqliteItemStorage[GraphExecutionState]( graph_execution_manager = SqliteItemStorage[GraphExecutionState](
filename=db_location, table_name="graph_executions" filename=db_location, table_name="graph_executions"

1
invokeai/app/cli_app.py
View File

@ -220,6 +220,7 @@ def invoke_cli():
db_location = ":memory:" db_location = ":memory:"
else: else:
db_location = config.db_path db_location = config.db_path
db_location.parent.mkdir(parents=True,exist_ok=True)
logger.info(f'InvokeAI database location is "{db_location}"') logger.info(f'InvokeAI database location is "{db_location}"')

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

@ -321,10 +321,6 @@ class ModelManager(object):
for name in sorted(self.config, key=str.casefold): for name in sorted(self.config, key=str.casefold):
stanza = self.config[name] stanza = self.config[name]
with open('log.txt','a') as file:
print(f'DEBUG: name={name}; stanza = {stanza}',file=file)
# don't include VAEs in listing (legacy style) # don't include VAEs in listing (legacy style)
if "config" in stanza and "/VAE/" in stanza["config"]: if "config" in stanza and "/VAE/" in stanza["config"]:
continue continue
@ -1355,8 +1351,6 @@ class ModelManager(object):
short_names = OmegaConf.load(Dataset_path).get('lora') or {} short_names = OmegaConf.load(Dataset_path).get('lora') or {}
for name in model_names: for name in model_names:
print(name)
name = short_names.get(name) or name name = short_names.get(name) or name
# HuggingFace style LoRA # HuggingFace style LoRA