Skip to content

Commit 2f43843

Browse files
committed
(fix) update logic for installing specific version
1 parent bbeb5cb commit 2f43843

File tree

1 file changed

+8
-3
lines changed

1 file changed

+8
-3
lines changed

invokeai/frontend/install/invokeai_update.py

Lines changed: 8 additions & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -58,7 +58,7 @@ def get_pypi_versions():
5858
versions.sort(key=LooseVersion, reverse=True)
5959
latest_version = [v for v in versions if "rc" not in v][0]
6060
latest_release_candidate = [v for v in versions if "rc" in v][0]
61-
return latest_version, latest_release_candidate
61+
return latest_version, latest_release_candidate, versions
6262

6363

6464
def welcome(latest_release: str, latest_prerelease: str):
@@ -107,7 +107,7 @@ def main():
107107
input("Press any key to continue...")
108108
return
109109

110-
latest_release, latest_prerelease = get_pypi_versions()
110+
latest_release, latest_prerelease, versions = get_pypi_versions()
111111

112112
welcome(latest_release, latest_prerelease)
113113

@@ -119,7 +119,12 @@ def main():
119119
elif choice == "2":
120120
release = latest_prerelease
121121
elif choice == "3":
122-
release = Prompt.ask("Enter an InvokeAI version name")
122+
while True:
123+
release = Prompt.ask("Enter an InvokeAI version")
124+
release.strip()
125+
if release in versions:
126+
break
127+
print(f":exclamation: [bold red]'{release}' is not a recognized InvokeAI release.[/red bold]")
123128

124129
extras = get_extras()
125130

0 commit comments

Comments
 (0)