|
def return_task_name(): |
|
return ('text2text-generation', "text-generation") |
|
|
|
|
|
def return_models(task_name): |
|
if task_name == "text2text-generation": |
|
models_parent = ( |
|
'google', |
|
"facebook", |
|
"microsoft", |
|
"langboat", |
|
"bloom", |
|
"allenai", |
|
"mbzuai", |
|
"lmsys", |
|
"starmpcc", |
|
"haining", |
|
"kaludi" |
|
|
|
) |
|
else: |
|
models_parent = ( |
|
'google', |
|
"facebook", |
|
"microsoft", |
|
"langboat", |
|
"databricks", |
|
"aisquared", |
|
"bloom", |
|
"allenai", |
|
"tiiuae", |
|
"openlm", |
|
"stabilityai", |
|
"eleutherai", |
|
"mbzuai", |
|
"cerebras", |
|
"open_assistant", |
|
"nomic_ai", |
|
"blinkdl", |
|
"lmsys", |
|
"together_computer", |
|
"mosaic_ml", |
|
"h20ai", |
|
) |
|
return models_parent |
|
|
|
def return_text2text_generation_models(model_parent, count=False): |
|
google_models_list = ( |
|
"google/flan-t5-small", |
|
"google/flan-t5-base", |
|
'google/flan-t5-large', |
|
'google/flan-t5-xl', |
|
'google/flan-t5-xxl', |
|
"google/byt5-xxl", |
|
"google/byt5-xl", |
|
"google/byt5-large", |
|
"google/byt5-small", |
|
"google/byt5-base", |
|
"google/mt5-xxl", |
|
"google/mt5-xl", |
|
"google/mt5-large", |
|
"google/mt5-small", |
|
"google/long-t5-local-base", |
|
"google/long-t5-local-large", |
|
"google/long-t5-tglobal-base", |
|
"google/long-t5-tglobal-large", |
|
"google/pegasus-big_patent", |
|
"google/pegasus-x-base", |
|
"google/pegasus-x-large", |
|
"google/pegasus-x-base-arxiv", |
|
"google/roberta2roberta_L-24_wikisplit", |
|
"google/roberta2roberta_L-24_discofuse", |
|
"google/switch-base-8", |
|
"google/switch-base-16", |
|
"google/switch-base-32", |
|
"google/switch-base-64", |
|
"google/switch-base-128", |
|
"google/switch-base-256", |
|
"google/switch-large-128", |
|
"google/switch-xxl-128", |
|
"google/switch-c-2048", |
|
"google/t5-11b-ssm", |
|
"google/t5-11b-ssm-nq", |
|
"google/t5-11b-ssm-nqo", |
|
"google/t5-11b-ssm-tqa", |
|
"google/t5-11b-ssm-tqao", |
|
"google/t5-11b-ssm-wq", |
|
"google/t5-11b-ssm-wqo", |
|
"google/t5-3b-ssm", |
|
"google/t5-3b-ssm-nq", |
|
"google/t5-3b-ssm-nqo", |
|
"google/t5-base-lm-adapt", |
|
"google/t5-base-lm-adapt", |
|
"google/t5-efficient-base", |
|
"google/t5-efficient-base-dl2", |
|
"google/t5-efficient-base-dl4", |
|
"google/t5-efficient-base-dl6", |
|
"google/t5-efficient-base-dl8", |
|
"google/t5-efficient-base-dm256", |
|
"google/t5-efficient-base-dm512", |
|
"google/t5-efficient-base-dm1000", |
|
"google/t5-efficient-base-dm2000", |
|
"google/t5-efficient-base-el2", |
|
"google/t5-efficient-base-el4", |
|
"google/t5-efficient-base-el6", |
|
"google/t5-efficient-base-el8", |
|
"google/t5-efficient-base-el16", |
|
"google/t5-efficient-base-nl40", |
|
"google/t5-efficient-base-nl48", |
|
"google/t5-efficient-base-nl8", |
|
"google/t5-efficient-large", |
|
"google/t5-efficient-large-dl12", |
|
"google/t5-efficient-large-dl16", |
|
"google/t5-efficient-large-dl2", |
|
"google/t5-efficient-large-dl32", |
|
"google/t5-efficient-large-dl4", |
|
"google/t5-efficient-large-dl6", |
|
"google/t5-efficient-large-dl8", |
|
"google/t5-efficient-large-dm128", |
|
"google/t5-efficient-large-dm2000", |
|
"google/t5-efficient-large-dm256", |
|
"google/t5-efficient-large-dm512", |
|
"google/t5-efficient-large-dm768", |
|
"google/t5-efficient-large-el12", |
|
"google/t5-efficient-large-el2", |
|
"google/t5-efficient-large-el4", |
|
"google/t5-efficient-large-el6", |
|
"google/t5-efficient-large-el8", |
|
"google/t5-efficient-large-kv128", |
|
"google/t5-efficient-large-kv16", |
|
"google/t5-efficient-large-kv256", |
|
"google/t5-efficient-large-kv32", |
|
"google/t5-efficient-large-nh12", |
|
"google/t5-efficient-large-nh2", |
|
"google/t5-efficient-large-nh24", |
|
"google/t5-efficient-large-nh32", |
|
"google/t5-efficient-large-nh4", |
|
"google/t5-efficient-large-nh8", |
|
"google/t5-efficient-large-nh8-nl32", |
|
"google/t5-efficient-large-nl10", |
|
"google/t5-efficient-large-nl12", |
|
"google/t5-efficient-large-nl16", |
|
"google/t5-efficient-large-nl2", |
|
"google/t5-efficient-large-nl20", |
|
"google/t5-efficient-large-nl32", |
|
"google/t5-efficient-large-nl36", |
|
"google/t5-efficient-large-nl4", |
|
"google/t5-efficient-large-nl8", |
|
"google/t5-efficient-mini", |
|
"google/t5-efficient-mini-nl12", |
|
"google/t5-efficient-mini-nl24", |
|
"google/t5-efficient-mini-nl6", |
|
"google/t5-efficient-mini-nl8", |
|
"google/t5-efficient-small", |
|
"google/t5-efficient-small-dl12", |
|
"google/t5-efficient-small-dl16", |
|
"google/t5-efficient-small-dl2", |
|
"google/t5-efficient-small-dl4", |
|
"google/t5-efficient-small-dl8", |
|
"google/t5-efficient-small-dm1000", |
|
"google/t5-efficient-small-dm128", |
|
"google/t5-efficient-small-dm2000", |
|
"google/t5-efficient-small-dm256", |
|
"google/t5-efficient-small-dm768", |
|
"google/t5-efficient-small-el12", |
|
"google/t5-efficient-small-el16", |
|
"google/t5-efficient-small-el16-dl1", |
|
"google/t5-efficient-small-el16-dl2", |
|
"google/t5-efficient-small-el16-dl4", |
|
"google/t5-efficient-small-el16-dl8", |
|
"google/t5-efficient-small-el2", |
|
"google/t5-efficient-small-el32", |
|
"google/t5-efficient-small-el4", |
|
"google/t5-efficient-small-el48", |
|
"google/t5-efficient-small-el64", |
|
"google/t5-efficient-small-el8", |
|
"google/t5-efficient-small-el8-dl1", |
|
"google/t5-efficient-small-el8-dl2", |
|
"google/t5-efficient-small-el8-dl4", |
|
"google/t5-efficient-small-ff1000", |
|
"google/t5-efficient-small-ff12000", |
|
"google/t5-efficient-small-ff3000", |
|
"google/t5-efficient-small-ff6000", |
|
"google/t5-efficient-small-ff9000", |
|
"google/t5-efficient-small-kv128", |
|
"google/t5-efficient-small-kv16", |
|
"google/t5-efficient-small-kv256", |
|
"google/t5-efficient-small-kv32", |
|
"google/t5-efficient-small-nl16", |
|
"google/t5-efficient-small-nl2", |
|
"google/t5-efficient-small-nl20", |
|
"google/t5-efficient-small-nl22", |
|
"google/t5-efficient-small-nl24", |
|
"google/t5-efficient-small-nl32", |
|
"google/t5-efficient-small-nl36", |
|
"google/t5-efficient-small-nl4", |
|
"google/t5-efficient-small-nl40", |
|
"google/t5-efficient-small-nl48", |
|
"google/t5-efficient-small-nl8", |
|
"google/t5-efficient-tiny", |
|
"google/t5-efficient-tiny-dl2", |
|
"google/t5-efficient-tiny-dl6", |
|
"google/t5-efficient-tiny-dl8", |
|
"google/t5-efficient-tiny-el12", |
|
"google/t5-efficient-tiny-el2", |
|
"google/t5-efficient-tiny-el6", |
|
"google/t5-efficient-tiny-el8", |
|
"google/t5-efficient-tiny-ff12000", |
|
"google/t5-efficient-tiny-ff2000", |
|
"google/t5-efficient-tiny-ff3000", |
|
"google/t5-efficient-tiny-ff6000", |
|
"google/t5-efficient-tiny-ff9000", |
|
"google/t5-efficient-tiny-nh1", |
|
"google/t5-efficient-tiny-nh16", |
|
"google/t5-efficient-tiny-nh32", |
|
"google/t5-efficient-tiny-nh8", |
|
"google/t5-efficient-tiny-nl12", |
|
"google/t5-efficient-tiny-nl16", |
|
"google/t5-efficient-tiny-nl2", |
|
"google/t5-efficient-tiny-nl24", |
|
"google/t5-efficient-tiny-nl32", |
|
"google/t5-efficient-tiny-nl6", |
|
"google/t5-efficient-tiny-nl8", |
|
"google/t5-efficient-xl", |
|
"google/t5-efficient-xl-nl12", |
|
"google/t5-efficient-xl-nl16", |
|
"google/t5-efficient-xl-nl2", |
|
"google/t5-efficient-xl-nl28", |
|
"google/t5-efficient-xl-nl4", |
|
"google/t5-efficient-xl-nl6", |
|
"google/t5-efficient-xl-nl8", |
|
"google/t5-efficient-xxl", |
|
"google/t5-efficient-xxl-nl4", |
|
"google/t5-large-lm-adapt", |
|
"google/t5-large-ssm", |
|
"google/t5-large-ssm-nq", |
|
"google/t5-large-ssm-nqo", |
|
"google/t5-small-lm-adapt", |
|
"google/t5-small-ssm", |
|
"google/t5-small-ssm-nq", |
|
"google/t5-v1_1-base", |
|
"google/t5-v1_1-large", |
|
"google/t5-v1_1-small", |
|
"google/t5-v1_1-xl", |
|
"google/t5-v1_1-xxl", |
|
"google/t5-xl-lm-adapt", |
|
"google/t5-xl-ssm-nq", |
|
"google/t5-xxl-lm-adapt", |
|
"google/t5-xxl-ssm", |
|
"google/t5-xxl-ssm-nq", |
|
"google/t5-xxl-ssm-nqo", |
|
"google/t5-xxl-ssm-tqa", |
|
"google/t5-xxl-ssm-tqao", |
|
"google/t5-xxl-ssm-wq", |
|
"google/t5-xxl-ssm-wqo", |
|
"google/t5_11b_trueteacher_and_anli", |
|
"google/ul2", |
|
"google/umt5-base", |
|
"google/umt5-small", |
|
"google/umt5-xl", |
|
"google/umt5-xxl", |
|
) |
|
|
|
mbzuai_models_list = ( |
|
"MBZUAI/LaMini-Flan-T5-783M", |
|
"MBZUAI/LaMini-Flan-T5-248M", |
|
"MBZUAI/LaMini-Flan-T5-77M", |
|
"MBZUAI/LaMini-T5-738M", |
|
"MBZUAI/LaMini-T5-223M", |
|
"MBZUAI/LaMini-T5-61M", |
|
) |
|
bloom_models_list = ( |
|
"bigscience/T0_3B", |
|
"bigscience/T0_original_task_only", |
|
"bigscience/T0_single_prompt", |
|
"bigscience/T0p", |
|
"bigscience/T0", |
|
"bigscience/T0pp", |
|
"bigscience/mt0-xxl-p3", |
|
"bigscience/mt0-xxl", |
|
"bigscience/mt0-large", |
|
"bigscience/mt0-base", |
|
"bigscience/mt0-small", |
|
"bigscience/mt0-xxl-mt", |
|
) |
|
|
|
lmsys_models_list = ( |
|
"lmsys/fastchat-t5-3b-v1.0", |
|
) |
|
|
|
facebook_models_list = ( |
|
'facebook/mbart-large-50-many-to-many-mmt', |
|
'facebook/musicgen-small', |
|
'facebook/musicgen-medium', |
|
"facebook/musicgen-large", |
|
'facebook/m2m100_418M', |
|
'facebook/mbart-large-50-one-to-many-mmt', |
|
'facebook/mbart-large-50-many-to-one-mmt', |
|
'facebook/mbart-large-50', |
|
'facebook/mgenre-wiki', |
|
'facebook/genre-linking-aidayago2', |
|
'facebook/genre-linking-blink', |
|
'facebook/genre-kilt', |
|
'facebook/m2m100-12B-avg-10-ckpt', |
|
'facebook/m2m100-12B-avg-5-ckpt', |
|
'facebook/m2m100-12B-last-ckpt', |
|
'facebook/m2m100_1.2B' |
|
) |
|
|
|
microsoft_models_list = ( |
|
"microsoft/prophetnet-large-uncased", |
|
"microsoft/codereviewer", |
|
"microsoft/xprophetnet-large-wiki100-cased", |
|
"microsoft/xprophetnet-large-wiki100-cased-xglue-qg", |
|
"microsoft/xprophetnet-large-wiki100-cased-xglue-ntg", |
|
"microsoft/prophetnet-large-uncased-squad-qg", |
|
"microsoft/prophetnet-large-uncased-cnndm", |
|
) |
|
starmpcc_models_list = ( |
|
"starmpcc/Asclepius-13B", |
|
"starmpcc/Asclepius-7B" |
|
) |
|
|
|
langboat_models_list = ( |
|
"Langboat/mengzi-t5-base", |
|
"Langboat/mengzi-t5-base-mt" |
|
) |
|
|
|
haining_models_list = ( |
|
"haining/scientific_abstract_simplification", |
|
"haining/poem_interpretation_allpoetry169k_baseline", |
|
"haining/poem_interpretation_allpoetry169k_full", |
|
"haining/lyrics_interpretation_nonnegative", |
|
"haining/poem_interpretation_allpoetry169k", |
|
"haining/sas_baseline", |
|
) |
|
|
|
kaludi_models_list = ( |
|
"Kaludi/chatgpt-gpt4-prompts-bart-large-cnn-samsum", |
|
"Kaludi/Customer-Support-Assistant", |
|
"Kaludi/Customer-Support-Assistant-V2", |
|
"Kaludi/historical-events-reimagined" |
|
) |
|
|
|
allenai_models_list = ( |
|
'allenai/macaw-large', |
|
'allenai/entailer-11b', |
|
'allenai/entailer-large', |
|
'allenai/System4_classify_FigLang2022', |
|
'allenai/System4_explain_FigLang2022', |
|
'allenai/System3_DREAM_FLUTE_all_dimensions_FigLang2022', |
|
'allenai/System3_DREAM_FLUTE_social_norm_FigLang2022', |
|
'allenai/System3_DREAM_FLUTE_consequence_FigLang2022', |
|
'allenai/System3_DREAM_FLUTE_motivation_FigLang2022', |
|
'allenai/System3_DREAM_FLUTE_emotion_FigLang2022', |
|
'allenai/System2_FigLang2022', |
|
'allenai/System1_FigLang2022', |
|
'allenai/DREAM', |
|
'allenai/led-base-16384-cochrane', |
|
'allenai/bart-large-multi_lexsum-long-multitask', |
|
'allenai/bart-large-multi_lexsum-source-multitask', |
|
'allenai/pegasus-multi_lexsum-short-tiny', |
|
'allenai/pegasus-multi_lexsum-long-tiny', |
|
'allenai/pegasus-multi_lexsum-long-short', |
|
'allenai/bart-large-multi_lexsum-short-tiny', |
|
'allenai/bart-large-multi_lexsum-long-tiny', |
|
'allenai/bart-large-multi_lexsum-long-short', |
|
'allenai/primera-multi_lexsum-source-tiny', |
|
'allenai/primera-multi_lexsum-source-short', |
|
'allenai/primera-multi_lexsum-source-long', |
|
'allenai/led-base-16384-multi_lexsum-source-tiny', |
|
'allenai/led-base-16384-multi_lexsum-source-short', |
|
'allenai/led-base-16384-multi_lexsum-source-long', |
|
'allenai/led-base-16384-ms2', |
|
'allenai/mtk-instruct-11b-def-pos', |
|
'allenai/tk-instruct-small-def-pos', |
|
'allenai/tk-instruct-base-def-pos', |
|
'allenai/tk-instruct-large-def-pos', |
|
'allenai/mtk-instruct-3b-def-pos', |
|
'allenai/tk-instruct-3b-def-pos-neg-expl', |
|
'allenai/tk-instruct-3b-def-pos-neg', |
|
'allenai/tk-instruct-3b-pos', |
|
'allenai/tk-instruct-3b-def-pos', |
|
'allenai/tk-instruct-3b-def', |
|
'allenai/tk-instruct-11b-def-pos-neg-expl', |
|
'allenai/tk-instruct-11b-def-pos', |
|
'allenai/tk-instruct-11b-def', |
|
'allenai/PRIMERA-arxiv', |
|
'allenai/PRIMERA-wcep', |
|
'allenai/PRIMERA-multixscience', |
|
'allenai/PRIMERA-multinews', |
|
'allenai/PRIMERA', |
|
'allenai/unifiedqa-v2-t5-small-1363200', |
|
'allenai/unifiedqa-v2-t5-small-1251000', |
|
'allenai/unifiedqa-v2-t5-large-1363200', |
|
'allenai/unifiedqa-v2-t5-large-1251000', |
|
'allenai/unifiedqa-v2-t5-base-1363200', |
|
'allenai/unifiedqa-v2-t5-base-1251000', |
|
'allenai/unifiedqa-v2-t5-3b-1363200', |
|
'allenai/unifiedqa-v2-t5-3b-1251000', |
|
'allenai/unifiedqa-v2-t5-11b-1363200', |
|
'allenai/unifiedqa-v2-t5-11b-1251000', |
|
'allenai/unifiedqa-t5-small', |
|
'allenai/unifiedqa-t5-large', |
|
'allenai/unifiedqa-t5-base', |
|
'allenai/unifiedqa-t5-3b', |
|
'allenai/unifiedqa-t5-11b', |
|
'allenai/tailor', |
|
'allenai/t5-small-squad2-question-generation', |
|
'allenai/t5-small-squad2-next-word-generator-squad', |
|
'allenai/t5-small-squad11', |
|
'allenai/t5-small-next-word-generator-qoogle', |
|
'allenai/macaw-answer-11b', |
|
'allenai/macaw-3b', |
|
'allenai/macaw-11b', |
|
'allenai/led-large-16384', |
|
'allenai/led-large-16384-arxiv', |
|
'allenai/led-base-16384' |
|
) |
|
|
|
model_dict = { |
|
"google": google_models_list, |
|
"mbzuai": mbzuai_models_list, |
|
"bloom": bloom_models_list, |
|
"lmsys": lmsys_models_list, |
|
"facebook": facebook_models_list, |
|
"microsoft": microsoft_models_list, |
|
"starmpcc": starmpcc_models_list, |
|
"langboat": langboat_models_list, |
|
"haining": haining_models_list, |
|
"kaludi": kaludi_models_list, |
|
"allenai": allenai_models_list |
|
} |
|
if count is True: |
|
models_count = 0 |
|
for i in model_dict: |
|
models_count += len(model_dict[i]) |
|
return models_count |
|
|
|
return model_dict[model_parent] |
|
|
|
|
|
|
|
def return_text_generation_models(model_parent, count=False): |
|
google_models_list = ( |
|
"google/reformer-enwik8", |
|
"google/reformer-crime-and-punishment", |
|
) |
|
mbzuai_models_list = ( |
|
"MBZUAI/LaMini-Cerebras-111M", |
|
"MBZUAI/LaMini-Cerebras-256M", |
|
"MBZUAI/LaMini-Cerebras-590M", |
|
"MBZUAI/LaMini-Cerebras-1.3B", |
|
"MBZUAI/LaMini-GPT-774M", |
|
"MBZUAI/LaMini-GPT-124M", |
|
"MBZUAI/LaMini-GPT-1.5B", |
|
"MBZUAI/LaMini-Neo-125M", |
|
"MBZUAI/LaMini-Neo-1.3B", |
|
) |
|
|
|
eleutherai_models_list=( |
|
"EleutherAI/pythia-14m", |
|
"EleutherAI/pythia-31m", |
|
"EleutherAI/pythia-1b-deduped", |
|
"EleutherAI/pythia-2.8b-v0", |
|
"EleutherAI/pythia-1b-v0", |
|
"EleutherAI/pythia-410m-v0", |
|
"EleutherAI/pythia-70m-deduped-v0", |
|
"EleutherAI/pythia-2.8b-deduped-v0", |
|
"EleutherAI/pythia-1b-deduped-v0", |
|
"EleutherAI/pythia-410m-deduped-v0", |
|
"EleutherAI/pythia-160m-deduped-v0", |
|
"EleutherAI/pythia-6.9b-deduped-v0", |
|
"EleutherAI/pythia-70m-deduped", |
|
"EleutherAI/pythia-70m", |
|
"EleutherAI/pythia-2.8b-deduped", |
|
"EleutherAI/pythia-1b", |
|
"EleutherAI/pythia-410m-deduped", |
|
"EleutherAI/pythia-160m-deduped", |
|
"EleutherAI/pythia-160m-v0", |
|
"EleutherAI/pythia-1.4b-deduped-v0", |
|
"EleutherAI/pythia-1.4b", |
|
"EleutherAI/pythia-410m", |
|
"EleutherAI/pythia-intervention-410m-deduped", |
|
"EleutherAI/gpt-neo-125m", |
|
"EleutherAI/gpt-neo-2.7B", |
|
"EleutherAI/gpt-neo-1.3B", |
|
"EleutherAI/pythia-160m", |
|
"EleutherAI/gpt-neox-20b", |
|
"EleutherAI/gpt-j-6b", |
|
"EleutherAI/pythia-2.8b", |
|
"EleutherAI/pythia-12b-deduped", |
|
"EleutherAI/pythia-6.9b-deduped", |
|
"EleutherAI/pythia-1.4b-deduped", |
|
"EleutherAI/pythia-12b", |
|
"EleutherAI/pythia-6.9b", |
|
"EleutherAI/polyglot-ko-12.8b", |
|
"EleutherAI/polyglot-ko-5.8b", |
|
"EleutherAI/polyglot-ko-3.8b", |
|
"EleutherAI/polyglot-ko-1.3b", |
|
"EleutherAI/pythia-intervention-6.9b-deduped", |
|
"EleutherAI/pythia-intervention-1.4b-deduped", |
|
"EleutherAI/pythia-intervention-70m-deduped", |
|
"EleutherAI/pythia-intervention-long-1.4b-deduped", |
|
"EleutherAI/pythia-70m-v0", |
|
"EleutherAI/pythia-1.4b-v0", |
|
"EleutherAI/pythia-6.9b-v0", |
|
"EleutherAI/pythia-12b-deduped-v0", |
|
"EleutherAI/pythia-12b-v0", |
|
"EleutherAI/pythia-160m-seed3", |
|
"EleutherAI/pythia-160m-seed2", |
|
"EleutherAI/pythia-160m-seed1", |
|
"EleutherAI/neox-ckpt-pythia-6.9b-deduped", |
|
"EleutherAI/pythia-160m-hiddendropout", |
|
"EleutherAI/pythia-160m-attndropout", |
|
"EleutherAI/pythia-160m-alldropout", |
|
"EleutherAI/pythia-6.9b-deduped-v0-seed42", |
|
) |
|
cerebras_models_list = ( |
|
"cerebras/btlm-3b-8k-base", |
|
"cerebras/cerebras-GPT-13B", |
|
"cerebras/cerebras-GPT-6.7B", |
|
"cerebras/cerebras-GPT-2.7B", |
|
"cerebras/cerebras-GPT-1.3B", |
|
"cerebras/cerebras-GPT-590M", |
|
"cerebras/cerebras-GPT-256M", |
|
"cerebras/cerebras-GPT-111M", |
|
) |
|
open_assistant_models_list = ( |
|
"OpenAssistant/codellama-13b-oasst-sft-v10", |
|
"OpenAssistant/llama2-70b-oasst-sft-v10", |
|
"OpenAssistant/llama2-13b-megacode2-oasst", |
|
"OpenAssistant/falcon-40b-megacode2-oasst", |
|
"OpenAssistant/pythia-12b-sft-v8-rlhf-2k-steps", |
|
"OpenAssistant/llama2-13b-orca-8k-3319", |
|
"OpenAssistant/falcon-7b-sft-mix-2000", |
|
"OpenAssistant/falcon-7b-sft-top1-696", |
|
"OpenAssistant/falcon-40b-sft-mix-1226", |
|
"OpenAssistant/falcon-40b-sft-top1-560", |
|
"OpenAssistant/pythia-12b-sft-v8-2.5k-steps", |
|
"OpenAssistant/pythia-12b-sft-v8-7k-steps", |
|
"OpenAssistant/pythia-12b-pre-v8-12.5k-steps", |
|
"OpenAssistant/oasst-sft-4-pythia-12b-epoch-3.5", |
|
"OpenAssistant/stablelm-7b-sft-v7-epoch-3", |
|
"OpenAssistant/oasst-sft-1-pythia-12b", |
|
"OpenAssistant/galactica-6.7b-finetuned", |
|
) |
|
|
|
databricks_models_list = ( |
|
|
|
"databricks/dolly-v2-7b", |
|
"databricks/dolly-v2-3b", |
|
"databricks/dolly-v2-12b", |
|
"databricks/dolly-v1-6b", |
|
) |
|
|
|
nomic_ai_models_list = ( |
|
"nomic-ai/gpt4all-falcon", |
|
"nomic-ai/gpt4all-j", |
|
"nomic-ai/gpt4all-mpt", |
|
"nomic-ai/gpt4all-13b-snoozy", |
|
"nomic-ai/gpt4all-j-lora", |
|
) |
|
|
|
blinkdl_models_list = ( |
|
"BlinkDL/rwkv-5-world", |
|
"BlinkDL/rwkv-4-world", |
|
"BlinkDL/rwkv-4-raven", |
|
"BlinkDL/rwkv-4-pile-7b", |
|
"BlinkDL/rwkv-4-pile-14b", |
|
"BlinkDL/rwkv-4-novel", |
|
"BlinkDL/rwkv-4-pileplus", |
|
"BlinkDL/rwkv-4-pile-430m", |
|
"BlinkDL/rwkv-4-pile-3b", |
|
"BlinkDL/rwkv-4-pile-1b5", |
|
"BlinkDL/rwkv-4-pile-169m", |
|
"BlinkDL/rwkv-3-pile-1b5", |
|
"BlinkDL/rwkv-3-pile-430m", |
|
"BlinkDL/rwkv-2-pile-430m", |
|
"BlinkDL/rwkv-3-pile-169m", |
|
) |
|
|
|
ai_squared_models_list = ( |
|
"aisquared/dlite-dais-2023", |
|
"aisquared/chopt-1_3b", |
|
"aisquared/chopt-350m", |
|
"aisquared/chopt-125m", |
|
"aisquared/chopt-2_7b", |
|
"aisquared/dlite-v2-1_5b", |
|
"aisquared/dlite-v2-774m", |
|
"aisquared/dlite-v2-355m", |
|
"aisquared/dlite-v2-124m", |
|
"aisquared/dlite-v1-355m", |
|
"aisquared/dlite-v1-774m", |
|
"aisquared/dlite-v1-1_5b", |
|
"aisquared/dlite-v1-124m", |
|
"aisquared/chopt-research-350m", |
|
"aisquared/chopt-research-125m", |
|
"aisquared/chopt-research-2_7b", |
|
"aisquared/chopt-research-1_3b", |
|
) |
|
|
|
bloom_models_list = ( |
|
"bigscience/bloom-3b-intermediate", |
|
"bigscience/bloom", |
|
"bigscience/bloomz-p3", |
|
"bigscience/bloomz-mt", |
|
"bigscience/bloomz-7b1-mt", |
|
"bigscience/bloom-1b7-intermediate", |
|
"bigscience/bloom-560m-intermediate", |
|
"bigscience/bloomz-560m", |
|
"bigscience/bloomz-1b1", |
|
"bigscience/bloomz-1b7", |
|
"bigscience/bloomz-3b", |
|
"bigscience/bloomz-7b1", |
|
'bigscience/bloomz', |
|
"bigscience/bloom-1b7", |
|
"bigscience/bloom-560m", |
|
"bigscience/bloom-3b", |
|
"bigscience/bigscience-small-testing", |
|
"bigscience/distill-bloom-1b3", |
|
"bigscience/bloom-1b1", |
|
"bigscience/distill-bloom-1b3-10x", |
|
"bigscience/test-bloomd", |
|
"bigscience/test-bloomd-6b3", |
|
"bigscience/bloom-7b1", |
|
"bigscience/bloom-petals", |
|
"bigscience/bloom-1b1-intermediate", |
|
"bigscience/bloom-7b1-intermediate", |
|
"bigscience/bloom-7b1-petals", |
|
"bigscience/bloomz-petals", |
|
"bigscience/bloomz-7b1-p3", |
|
) |
|
tiiuae_models_list = ( |
|
"tiiuae/falcon-180B", |
|
"tiiuae/falcon-180B-chat", |
|
"tiiuae/falcon-40b", |
|
"tiiuae/falcon-7b", |
|
"tiiuae/falcon-7b-instruct", |
|
"tiiuae/falcon-40b-instruct", |
|
"tiiuae/falcon-rw-7b", |
|
"tiiuae/falcon-rw-1b", |
|
) |
|
|
|
openlm_models_list = ( |
|
"openlm-research/open_llama_3b_v2", |
|
"openlm-research/open_llama_7b_v2", |
|
"openlm-research/open_llama_13b", |
|
"openlm-research/open_llama_7b", |
|
"openlm-research/open_llama_3b", |
|
) |
|
stabilityai_models_list = ( |
|
"stabilityai/StableBeluga-7B", |
|
"stabilityai/StableBeluga-13B", |
|
"stabilityai/StableBeluga2", |
|
"stabilityai/stablelm-base-alpha-3b-v2", |
|
"stabilityai/stablelm-base-alpha-7b-v2", |
|
"stabilityai/japanese-stablelm-instruct-alpha-7b", |
|
"stabilityai/japanese-stablelm-base-alpha-7b", |
|
"stabilityai/stablecode-completion-alpha-3b-4k", |
|
"stabilityai/stablecode-instruct-alpha-3b", |
|
"stabilityai/stablecode-completion-alpha-3b", |
|
"stabilityai/StableBeluga1-Delta", |
|
"stabilityai/stablelm-base-alpha-3b", |
|
"stabilityai/stablelm-base-alpha-7b", |
|
"stabilityai/stablelm-tuned-alpha-3b", |
|
"stabilityai/stablelm-tuned-alpha-7b", |
|
) |
|
|
|
lmsys_models_list = ( |
|
"lmsys/vicuna-13b-v1.5-16k", |
|
"lmsys/vicuna-13b-v1.5", |
|
"lmsys/vicuna-7b-v1.5-16k", |
|
"lmsys/longchat-7b-v1.5-32k", |
|
"lmsys/vicuna-7b-v1.5", |
|
"lmsys/vicuna-7b-v1.3", |
|
"lmsys/vicuna-13b-v1.3", |
|
"lmsys/vicuna-7b-v1.1", |
|
"lmsys/vicuna-13b-v1.1", |
|
"lmsys/vicuna-13b-delta-v0", |
|
"lmsys/vicuna-7b-delta-v0", |
|
"lmsys/vicuna-13b-delta-v1.1", |
|
"lmsys/vicuna-7b-delta-v1.1", |
|
"lmsys/vicuna-33b-v1.3", |
|
"lmsys/longchat-13b-16k", |
|
'lmsys/longchat-7b-16k', |
|
) |
|
|
|
togethercomputer_models_list = ( |
|
'togethercomputer/Llama-2-7B-32K-Instruct', |
|
'togethercomputer/RedPajama-INCITE-7B-Instruct', |
|
'togethercomputer/LLaMA-2-7B-32K', |
|
'togethercomputer/RedPajama-INCITE-7B-Base', |
|
'togethercomputer/RedPajama-INCITE-7B-Chat', |
|
'togethercomputer/RedPajama-INCITE-Chat-3B-v1', |
|
'togethercomputer/RedPajama-INCITE-Instruct-3B-v1', |
|
'togethercomputer/RedPajama-INCITE-Base-3B-v1', |
|
'togethercomputer/GPT-NeoXT-Chat-Base-20B', |
|
'togethercomputer/Pythia-Chat-Base-7B', |
|
'togethercomputer/GPT-JT-Moderation-6B', |
|
'togethercomputer/GPT-JT-6B-v1', |
|
'togethercomputer/GPT-JT-6B-v0' |
|
) |
|
mosaic_models_list = ( |
|
'mosaicml/mpt-7b-chat', |
|
'mosaicml/mpt-30b-chat', |
|
'mosaicml/mpt-7b-8k-chat', |
|
'mosaicml/mpt-7b-instruct', |
|
'mosaicml/mpt-7b-8k-instruct', |
|
'mosaicml/mpt-7b-8k', |
|
'mosaicml/mpt-30b', |
|
'mosaicml/mpt-7b', |
|
'mosaicml/mpt-7b-storywriter', |
|
'mosaicml/mpt-30b-instruct', |
|
'mosaicml/mpt-1b-redpajama-200b', |
|
'mosaicml/mpt-1b-redpajama-200b-dolly' |
|
) |
|
|
|
h20ai_models_list = ( |
|
'h2oai/h2ogpt-16k-codellama-7b-python', |
|
'h2oai/h2ogpt-16k-codellama-7b-instruct', |
|
'h2oai/h2ogpt-16k-codellama-7b', |
|
'h2oai/h2ogpt-16k-codellama-34b-python', |
|
'h2oai/h2ogpt-16k-codellama-34b-instruct', |
|
'h2oai/h2ogpt-16k-codellama-13b-python', |
|
'h2oai/h2ogpt-16k-codellama-13b-instruct', |
|
'h2oai/h2ogpt-16k-codellama-13b', |
|
'h2oai/h2ogpt-16k-codellama-34b', |
|
'h2oai/h2ogpt-4096-llama2-13b-chat', |
|
'h2oai/h2ogpt-4096-llama2-70b-chat', |
|
'h2oai/h2ogpt-4096-llama2-7b-chat', |
|
'h2oai/h2ogpt-4096-llama2-13b', |
|
'h2oai/h2ogpt-4096-llama2-7b', |
|
'h2oai/h2ogpt-4096-llama2-70b', |
|
'h2oai/h2ogpt-gm-oasst1-en-2048-falcon-40b-v2', |
|
'h2oai/h2ogpt-gm-oasst1-en-2048-falcon-7b-v3', |
|
'h2oai/h2ogpt-research-oasst1-llama-65b', |
|
'h2oai/h2ogpt-gm-oasst1-en-xgen-7b-8k', |
|
'h2oai/h2ogpt-gm-oasst1-en-2048-open-llama-3b', |
|
'h2oai/h2ogpt-gm-oasst1-en-2048-open-llama-7b', |
|
'h2oai/h2ogpt-oasst1-falcon-40b', |
|
'h2oai/h2ogpt-gm-oasst1-en-2048-falcon-40b-v1', |
|
'h2oai/h2ogpt-gm-oasst1-en-2048-open-llama-13b', |
|
'h2oai/h2ogpt-oig-oasst1-falcon-40b', |
|
'h2oai/h2ogpt-gm-oasst1-en-2048-falcon-7b', |
|
'h2oai/h2ogpt-gm-oasst1-multilang-2048-falcon-7b', |
|
'h2oai/h2ogpt-oasst1-512-12b', |
|
'h2oai/h2ogpt-oig-oasst1-256-6_9b', |
|
'h2oai/h2ogpt-oig-oasst1-512-6_9b', |
|
'h2oai/h2ogpt-research-oig-oasst1-512-30b', |
|
'h2oai/h2ogpt-gm-oasst1-en-2048-open-llama-7b-preview-700bt', |
|
'h2oai/h2ogpt-gm-oasst1-en-1024-open-llama-7b-preview-400bt', |
|
'h2oai/h2ogpt-gm-oasst1-en-2048-open-llama-7b-preview-300bt-v2', |
|
'h2oai/h2ogpt-gm-oasst1-en-2048-open-llama-7b-preview-300bt', |
|
'h2oai/h2ogpt-gm-oasst1-en-1024-12b', |
|
'h2oai/h2ogpt-gm-oasst1-en-1024-20b', |
|
'h2oai/h2ogpt-gm-oasst1-multilang-1024-20b', |
|
'h2oai/h2ogpt-oasst1-512-20b' |
|
) |
|
|
|
facebook_models_list = ( |
|
'facebook/xglm-4.5B', |
|
'facebook/galactica-125m', |
|
'facebook/opt-iml-1.3b', |
|
'facebook/opt-iml-max-1.3b', |
|
'facebook/opt-iml-max-30b', |
|
'facebook/opt-iml-30b', |
|
'facebook/galactica-120b', |
|
'facebook/galactica-30b', |
|
'facebook/galactica-6.7b', |
|
'facebook/galactica-1.3b', |
|
'facebook/opt-66b', |
|
'facebook/opt-30b', |
|
'facebook/opt-13b', |
|
'facebook/opt-6.7b', |
|
'facebook/opt-2.7b', |
|
'facebook/opt-1.3b', |
|
'facebook/opt-350m', |
|
'facebook/opt-125m', |
|
'facebook/incoder-1B', |
|
'facebook/incoder-6B', |
|
'facebook/xglm-7.5B', |
|
'facebook/xglm-564M', |
|
'facebook/xglm-2.9B', |
|
'facebook/xglm-1.7B' |
|
) |
|
|
|
microsoft_models_list = ( |
|
"microsoft/dolly-v2-7b-olive-optimized", |
|
"microsoft/lts-gpt2-sm", |
|
"microsoft/BioGPT-Large", |
|
"microsoft/BioGPT-Large-PubMedQA", |
|
"microsoft/biogpt", |
|
"microsoft/git-large-vatex", |
|
"microsoft/git-base-msrvtt-qa", |
|
"microsoft/git-base-vatex", |
|
"microsoft/Promptist", |
|
"microsoft/CodeGPT-small-py", |
|
"microsoft/CodeGPT-small-py-adaptedGPT2", |
|
"microsoft/CodeGPT-small-java", |
|
"microsoft/CodeGPT-small-java-adaptedGPT2", |
|
) |
|
|
|
langboat_models_list = ( |
|
"Langboat/bloom-1b4-zh", |
|
"Langboat/ReGPT-125M-200G", |
|
"Langboat/bloom-6b4-zh", |
|
"Langboat/bloom-2b5-zh", |
|
"Langboat/bloom-800m-zh", |
|
"Langboat/bloom-389m-zh", |
|
"Langboat/mengzi-gpt-neo-base", |
|
) |
|
|
|
allenai_models_list = ( |
|
'allenai/bhaskara', |
|
'allenai/open-instruct-pythia-6.9b-tulu', |
|
'allenai/open-instruct-opt-6.7b-tulu', |
|
'allenai/open-instruct-sni-13b', |
|
'allenai/open-instruct-flan-v2-13b', |
|
'allenai/open-instruct-dolly-13b', |
|
'allenai/open-instruct-self-instruct-13b', |
|
'allenai/open-instruct-self-instruct-7b', |
|
'allenai/open-instruct-oasst1-13b', |
|
'allenai/open-instruct-oasst1-7b', |
|
'allenai/open-instruct-sni-7b', |
|
'allenai/open-instruct-baize-13b', |
|
'allenai/open-instruct-baize-7b', |
|
'allenai/open-instruct-cot-13b', |
|
'allenai/open-instruct-cot-7b', |
|
'allenai/open-instruct-gpt4-alpaca-13b', |
|
'allenai/open-instruct-gpt4-alpaca-7b', |
|
'allenai/open-instruct-human-mix-7b', |
|
'allenai/open-instruct-human-mix-13b', |
|
'allenai/open-instruct-sharegpt-7b', |
|
'allenai/open-instruct-sharegpt-13b', |
|
'allenai/open-instruct-sharegpt-30b', |
|
'allenai/open-instruct-human-mix-30b', |
|
'allenai/tulu-7b', |
|
'allenai/tulu-13b', |
|
'allenai/tulu-30b', |
|
'allenai/open-instruct-code-alpaca-13b', |
|
'allenai/open-instruct-code-alpaca-7b', |
|
'allenai/open-instruct-unnatural-instructions-13b', |
|
'allenai/open-instruct-unnatural-instructions-7b', |
|
'allenai/open-instruct-stanford-alpaca-13b', |
|
'allenai/open-instruct-stanford-alpaca-7b', |
|
'allenai/open-instruct-flan-v2-7b', |
|
'allenai/open-instruct-dolly-7b', |
|
'allenai/tulu-65b', |
|
'allenai/open-instruct-sharegpt-65b', |
|
'allenai/open-instruct-human-mix-65b' |
|
) |
|
|
|
model_dict = { |
|
"google": google_models_list, |
|
"mbzuai": mbzuai_models_list, |
|
"eleutherai": eleutherai_models_list, |
|
"cerebras": cerebras_models_list, |
|
"open_assistant": open_assistant_models_list, |
|
"databricks": databricks_models_list, |
|
"nomic_ai": nomic_ai_models_list, |
|
"blinkdl": blinkdl_models_list, |
|
"aisquared": ai_squared_models_list, |
|
"bloom": bloom_models_list, |
|
"tiiuae": tiiuae_models_list, |
|
"openlm": openlm_models_list, |
|
"stabilityai": stabilityai_models_list, |
|
"lmsys": lmsys_models_list, |
|
"together_computer": togethercomputer_models_list, |
|
"mosaic_ml": mosaic_models_list, |
|
"h20ai": h20ai_models_list, |
|
"facebook": facebook_models_list, |
|
"microsoft": microsoft_models_list, |
|
"langboat": langboat_models_list, |
|
"allenai": allenai_models_list |
|
} |
|
|
|
if count is True: |
|
models_count = 0 |
|
for i in model_dict: |
|
models_count += len(model_dict[i]) |
|
return models_count |
|
|
|
return model_dict[model_parent] |
|
|
