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

fix(config): if no invokeai.yaml is found, create a default one

Browse Source 
This fixes an issue with `test_images.py`, which tests the bulk images routers and imports the whole FastAPI app. This triggers the config logic which fails on the test runner, because it has no `invokeai.yaml`.

Also probably just good for graceful fallback.
This commit is contained in:
psychedelicious 2024-03-12 09:16:04 +11:00
parent 3f8e2bfd18
commit d09f03ef25
1 changed files with 13 additions and 5 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

18
invokeai/app/services/config/config_default.py
View File

@ -206,22 +206,30 @@ class InvokeAIAppConfig(BaseSettings):
with open(dest_path, "w") as file: with open(dest_path, "w") as file:
meta_dict = {"meta": ConfigMeta().model_dump()} meta_dict = {"meta": ConfigMeta().model_dump()}
config_dict = self.model_dump(mode="json", exclude_unset=True, exclude_defaults=True) config_dict = self.model_dump(mode="json", exclude_unset=True, exclude_defaults=True)
file.write("# Internal metadata\n") file.write("# Internal metadata - do not edit:\n")
file.write(yaml.dump(meta_dict, sort_keys=False)) file.write(yaml.dump(meta_dict, sort_keys=False))
file.write("\n") file.write("\n")
file.write("# User settings\n") file.write("# Put user settings here:\n")
file.write(yaml.dump(config_dict, sort_keys=False)) if len(config_dict) > 0:
file.write(yaml.dump(config_dict, sort_keys=False))
def merge_from_file(self, source_path: Optional[Path] = None) -> None: def merge_from_file(self, source_path: Optional[Path] = None) -> None:
"""Read the config from the `invokeai.yaml` file, migrating it if necessary and merging it into the singleton config. """Read the config from the `invokeai.yaml` file, migrating it if necessary and merging it into the singleton config.
This function will write to the `invokeai.yaml` file if the config is migrated. This function will write to the `invokeai.yaml` file if the config is migrated.
If there is no `invokeai.yaml` file, one will be written.
Args: Args:
source_path: Path to the config file. If not provided, the default path is used. source_path: Path to the config file. If not provided, the default path is used.
""" """
config_from_file = load_and_migrate_config(source_path or self.init_file_path) path = source_path or self.init_file_path
self.update_config(config_from_file)
if not path.exists():
self.write_file(path)
else:
config_from_file = load_and_migrate_config(path)
self.update_config(config_from_file)
def parse_args(self) -> None: def parse_args(self) -> None:
"""Parse the CLI args and set the runtime root directory.""" """Parse the CLI args and set the runtime root directory."""