From a6ef2429fa5a23de0bb1a28e50361f282daca9a2 Mon Sep 17 00:00:00 2001 From: Lou Bernardi Date: Fri, 21 Apr 2023 11:54:50 -0400 Subject: [PATCH] Add "do not download" and "download from HF" to download-model.py (#1439) --- download-model.py | 10 +++++++--- 1 file changed, 7 insertions(+), 3 deletions(-) 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: