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) update logic for installing specific version

Browse Source
This commit is contained in:
Millun Atluri 2023-12-19 11:04:21 +11:00
parent bbeb5cb477
commit 2f438431bd
1 changed files with 8 additions and 3 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

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

@ -58,7 +58,7 @@ def get_pypi_versions():
versions.sort(key=LooseVersion, reverse=True) versions.sort(key=LooseVersion, reverse=True)
latest_version = [v for v in versions if "rc" not in v][0] latest_version = [v for v in versions if "rc" not in v][0]
latest_release_candidate = [v for v in versions if "rc" in v][0] latest_release_candidate = [v for v in versions if "rc" in v][0]
return latest_version, latest_release_candidate return latest_version, latest_release_candidate, versions
def welcome(latest_release: str, latest_prerelease: str): def welcome(latest_release: str, latest_prerelease: str):
@ -107,7 +107,7 @@ def main():
input("Press any key to continue...") input("Press any key to continue...")
return return
latest_release, latest_prerelease = get_pypi_versions() latest_release, latest_prerelease, versions = get_pypi_versions()
welcome(latest_release, latest_prerelease) welcome(latest_release, latest_prerelease)
@ -119,7 +119,12 @@ def main():
elif choice == "2": elif choice == "2":
release = latest_prerelease release = latest_prerelease
elif choice == "3": elif choice == "3":
release = Prompt.ask("Enter an InvokeAI version name") while True:
release = Prompt.ask("Enter an InvokeAI version")
release.strip()
if release in versions:
break
print(f":exclamation: [bold red]'{release}' is not a recognized InvokeAI release.[/red bold]")
extras = get_extras() extras = get_extras()