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

feat: add force-reinstall option to the updater

Browse Source
This commit is contained in:
Eugene Brodsky 2024-01-25 01:24:33 -05:00
parent 701f14c1e3
commit d4ed64df7d
1 changed files with 10 additions and 0 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

10
invokeai/frontend/install/invokeai_update.py
View File

@ -182,9 +182,19 @@ def main():
print(f":exclamation: [bold red]'{release}' is not a recognized InvokeAI release.[/red bold]") print(f":exclamation: [bold red]'{release}' is not a recognized InvokeAI release.[/red bold]")
extras = get_extras() extras = get_extras()
console.line()
force_reinstall = Confirm.ask(
"[bold]Force reinstallation of all dependencies?[/] This [i]may[/] help fix a broken upgrade, but is usually not necessary.",
default=False,
)
console.line()
flags = [] flags = []
if (index_url := get_torch_extra_index_url()) is not None: if (index_url := get_torch_extra_index_url()) is not None:
flags.append(f"--extra-index-url {index_url}") flags.append(f"--extra-index-url {index_url}")
if force_reinstall:
flags.append("--force-reinstall")
flags = " ".join(flags) flags = " ".join(flags)
print(f":crossed_fingers: Upgrading to [yellow]{release}[/yellow]") print(f":crossed_fingers: Upgrading to [yellow]{release}[/yellow]")