Mirrors/text-generation-webui
1
0
Fork 0
mirror of https://github.com/oobabooga/text-generation-webui.git synced 2024-11-25 09:19:23 +01:00
Code Issues Packages Projects Releases Wiki Activity

Update download-model.py

Browse Source
This commit is contained in:
oobabooga 2023-04-01 14:47:47 -03:00 committed by GitHub
parent 4b57bd0d99
commit 526d5725db
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23
1 changed files with 12 additions and 11 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

23
download-model.py
View File

@ -63,16 +63,17 @@ def sanitize_branch_name(branch_name):
def select_model_from_default_options(): def select_model_from_default_options():
models = { models = {
"Pygmalion 6B original": ("PygmalionAI", "pygmalion-6b", "b8344bb4eb76a437797ad3b19420a13922aaabe1"), "OPT 6.7B": ("facebook", "opt-6.7b", "main"),
"Pygmalion 6B main": ("PygmalionAI", "pygmalion-6b", "main"), "OPT 2.7B": ("facebook", "opt-2.7b", "main"),
"Pygmalion 6B dev": ("PygmalionAI", "pygmalion-6b", "dev"), "OPT 1.3B": ("facebook", "opt-1.3b", "main"),
"Pygmalion 2.7B": ("PygmalionAI", "pygmalion-2.7b", "main"), "OPT 350M": ("facebook", "opt-350m", "main"),
"Pygmalion 1.3B": ("PygmalionAI", "pygmalion-1.3b", "main"), "GALACTICA 6.7B": ("facebook", "galactica-6.7b", "main"),
"Pygmalion 350m": ("PygmalionAI", "pygmalion-350m", "main"), "GALACTICA 1.3B": ("facebook", "galactica-1.3b", "main"),
"OPT 6.7b": ("facebook", "opt-6.7b", "main"), "GALACTICA 125M": ("facebook", "galactica-125m", "main"),
"OPT 2.7b": ("facebook", "opt-2.7b", "main"), "Pythia-6.9B-deduped": ("EleutherAI", "pythia-6.9b-deduped", "main"),
"OPT 1.3b": ("facebook", "opt-1.3b", "main"), "Pythia-2.8B-deduped": ("EleutherAI", "pythia-2.8b-deduped", "main"),
"OPT 350m": ("facebook", "opt-350m", "main"), "Pythia-1.4B-deduped": ("EleutherAI", "pythia-1.4b-deduped", "main"),
"Pythia-410M-deduped": ("EleutherAI", "pythia-410m-deduped", "main"),
} }
choices = {} choices = {}
@ -246,4 +247,4 @@ if __name__ == '__main__':
# Downloading the files # Downloading the files
print(f"Downloading the model to {output_folder}") print(f"Downloading the model to {output_folder}")
download_files(links, output_folder, args.threads) download_files(links, output_folder, args.threads)