AI/text-generation-webui
1
0
Fork 0
mirror of https://github.com/oobabooga/text-generation-webui.git synced 2024-10-01 01:26:03 -04:00
Code Issues Packages Projects Releases Wiki Activity

Add "do not download" and "download from HF" to download-model.py (#1439)

Browse Source
This commit is contained in:
Lou Bernardi 2023-04-21 11:54:50 -04:00 committed by GitHub
parent e1aa9d5173
commit a6ef2429fa
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23
1 changed files with 7 additions and 3 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

10
download-model.py
View File

@ -41,13 +41,17 @@ def select_model_from_default_options():
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 = chr(ord('A') + len(models)) char_hugging = chr(ord('A') + len(models))
print(f"{char}) None of the above") 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()
print("Input> ", end='') print("Input> ", end='')
choice = input()[0].strip().upper() 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. print("""\nThen type the name of your desired Hugging Face model in the format organization/name.
Examples: Examples: