Mirrors/text-generation-webui
1
0
Fork 0
mirror of https://github.com/oobabooga/text-generation-webui.git synced 2024-11-22 16:17:57 +01:00
Code Issues Packages Projects Releases Wiki Activity

Change a message in download-model.py

Browse Source
This commit is contained in:
oobabooga 2023-05-10 19:00:14 -03:00
parent b7a589afc8
commit c7ba2d4f3f
1 changed files with 3 additions and 3 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

6
download-model.py
View File

@ -34,25 +34,25 @@ def select_model_from_default_options():
"Pythia-1.4B-deduped": ("EleutherAI", "pythia-1.4b-deduped", "main"), "Pythia-1.4B-deduped": ("EleutherAI", "pythia-1.4b-deduped", "main"),
"Pythia-410M-deduped": ("EleutherAI", "pythia-410m-deduped", "main"), "Pythia-410M-deduped": ("EleutherAI", "pythia-410m-deduped", "main"),
} }
choices = {}
choices = {}
print("Select the model that you want to download:\n") print("Select the model that you want to download:\n")
for i, name in enumerate(models): for i, name in enumerate(models):
char = chr(ord('A') + i) char = chr(ord('A') + i)
choices[char] = name choices[char] = name
print(f"{char}) {name}") print(f"{char}) {name}")
char_hugging = chr(ord('A') + len(models)) char_hugging = chr(ord('A') + len(models))
print(f"{char_hugging}) Manually specify a Hugging Face model") print(f"{char_hugging}) Manually specify a Hugging Face model")
char_exit = chr(ord('A') + len(models) + 1) char_exit = chr(ord('A') + len(models) + 1)
print(f"{char_exit}) Do not download a model") print(f"{char_exit}) Do not download a model")
print() print()
print("Input> ", end='') print("Input> ", end='')
choice = input()[0].strip().upper() choice = input()[0].strip().upper()
if choice == char_exit: if choice == char_exit:
exit() exit()
elif choice == char_hugging: elif choice == char_hugging:
print("""\nThen type the name of your desired Hugging Face model in the format organization/name. print("""\nType the name of your desired Hugging Face model in the format organization/name.
Examples: Examples:
facebook/opt-1.3b facebook/opt-1.3b