File tree Expand file tree Collapse file tree 1 file changed +8
-3
lines changed
invokeai/frontend/install Expand file tree Collapse file tree 1 file changed +8
-3
lines changed Original file line number Diff line number Diff line change @@ -58,7 +58,7 @@ def get_pypi_versions():
58
58
versions .sort (key = LooseVersion , reverse = True )
59
59
latest_version = [v for v in versions if "rc" not in v ][0 ]
60
60
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
62
62
63
63
64
64
def welcome (latest_release : str , latest_prerelease : str ):
@@ -107,7 +107,7 @@ def main():
107
107
input ("Press any key to continue..." )
108
108
return
109
109
110
- latest_release , latest_prerelease = get_pypi_versions ()
110
+ latest_release , latest_prerelease , versions = get_pypi_versions ()
111
111
112
112
welcome (latest_release , latest_prerelease )
113
113
@@ -119,7 +119,12 @@ def main():
119
119
elif choice == "2" :
120
120
release = latest_prerelease
121
121
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]" )
123
128
124
129
extras = get_extras ()
125
130
You can’t perform that action at this time.
0 commit comments