diff --git a/download-model.py b/download-model.py index c35cf831..a095f398 100644 --- a/download-model.py +++ b/download-model.py @@ -41,13 +41,17 @@ def select_model_from_default_options(): char = chr(ord('A') + i) choices[char] = name print(f"{char}) {name}") - char = chr(ord('A') + len(models)) - print(f"{char}) None of the above") + char_hugging = chr(ord('A') + len(models)) + print(f"{char_hugging}) Manually specify a Hugging Face model") + char_exit = chr(ord('A') + len(models) + 1) + print(f"{char_exit}) Do not download a model") print() print("Input> ", end='') choice = input()[0].strip().upper() - if choice == char: + if choice == char_exit: + exit() + elif choice == char_hugging: print("""\nThen type the name of your desired Hugging Face model in the format organization/name. Examples: