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

Gracefully error without deleting invokeai.yaml

Browse Source
This commit is contained in:
Brandon Rising
2024-03-11 15:01:41 -04:00
committed by Mary Hipp Rogers
parent c454ccc65c
commit c74c1927ec
1 changed files with 14 additions and 2 deletions
Download Patch File Download Diff File Expand all files Collapse all files

16
invokeai/backend/install/invokeai_configure.py
View File

@ -17,7 +17,7 @@ import warnings
from argparse import Namespace from argparse import Namespace
from enum import Enum from enum import Enum
from pathlib import Path from pathlib import Path
from shutil import get_terminal_size from shutil import get_terminal_size, copy, move, rmtree
from typing import Any, Optional, Set, Tuple, Type, get_args, get_type_hints from typing import Any, Optional, Set, Tuple, Type, get_args, get_type_hints
from urllib import request from urllib import request
@ -929,6 +929,10 @@ def main() -> None:
errors = set() errors = set()
FORCE_FULL_PRECISION = opt.full_precision # FIXME global FORCE_FULL_PRECISION = opt.full_precision # FIXME global
new_init_file = config.root_path / "invokeai.yaml"
backup_init_file = new_init_file.with_suffix(".bak")
if new_init_file.exists():
copy(new_init_file, new_init_file.with_suffix(".bak"))
try: try:
# if we do a root migration/upgrade, then we are keeping previous # if we do a root migration/upgrade, then we are keeping previous
@ -943,7 +947,6 @@ def main() -> None:
install_helper = InstallHelper(config, logger) install_helper = InstallHelper(config, logger)
models_to_download = default_user_selections(opt, install_helper) models_to_download = default_user_selections(opt, install_helper)
new_init_file = config.root_path / "invokeai.yaml"
if opt.yes_to_all: if opt.yes_to_all:
write_default_options(opt, new_init_file) write_default_options(opt, new_init_file)
@ -975,8 +978,17 @@ def main() -> None:
input("Press any key to continue...") input("Press any key to continue...")
except WindowTooSmallException as e: except WindowTooSmallException as e:
logger.error(str(e)) logger.error(str(e))
if backup_init_file.exists():
move(backup_init_file, new_init_file)
except KeyboardInterrupt: except KeyboardInterrupt:
print("\nGoodbye! Come back soon.") print("\nGoodbye! Come back soon.")
if backup_init_file.exists():
move(backup_init_file, new_init_file)
except Exception:
print("An error occurred during installation.")
if backup_init_file.exists():
move(backup_init_file, new_init_file)
print(traceback.format_exc(), file=sys.stderr)
# ------------------------------------- # -------------------------------------