Spaces:
Runtime error
Runtime error
from dataclasses import dataclass | |
from enum import Enum | |
from typing import Dict | |
class ModelInfo: | |
name: str | |
symbol: str # emoji | |
class ModelType(Enum): | |
PT = ModelInfo(name="pretrained", symbol="🟢") | |
FT = ModelInfo(name="fine-tuned", symbol="🔶") | |
IFT = ModelInfo(name="instruction-tuned", symbol="⭕") | |
RL = ModelInfo(name="RL-tuned", symbol="🟦") | |
Unknown = ModelInfo(name="Unknown, add type to request file!", symbol="?") | |
def to_str(self, separator=" "): | |
return f"{self.value.symbol}{separator}{self.value.name}" | |
MODEL_TYPE_METADATA: Dict[str, ModelType] = { | |
"tiiuae/falcon-180B": ModelType.PT, | |
"Qwen/Qwen-7B": ModelType.PT, | |
"Qwen/Qwen-7B-Chat": ModelType.RL, | |
"notstoic/PygmalionCoT-7b": ModelType.IFT, | |
"aisquared/dlite-v1-355m": ModelType.IFT, | |
"aisquared/dlite-v1-1_5b": ModelType.IFT, | |
"aisquared/dlite-v1-774m": ModelType.IFT, | |
"aisquared/dlite-v1-124m": ModelType.IFT, | |
"aisquared/chopt-2_7b": ModelType.IFT, | |
"aisquared/dlite-v2-124m": ModelType.IFT, | |
"aisquared/dlite-v2-774m": ModelType.IFT, | |
"aisquared/dlite-v2-1_5b": ModelType.IFT, | |
"aisquared/chopt-1_3b": ModelType.IFT, | |
"aisquared/dlite-v2-355m": ModelType.IFT, | |
"augtoma/qCammel-13": ModelType.IFT, | |
"Aspik101/Llama-2-7b-hf-instruct-pl-lora_unload": ModelType.IFT, | |
"Aspik101/vicuna-7b-v1.3-instruct-pl-lora_unload": ModelType.IFT, | |
"TheBloke/alpaca-lora-65B-HF": ModelType.FT, | |
"TheBloke/tulu-7B-fp16": ModelType.IFT, | |
"TheBloke/guanaco-7B-HF": ModelType.FT, | |
"TheBloke/koala-7B-HF": ModelType.FT, | |
"TheBloke/wizardLM-7B-HF": ModelType.IFT, | |
"TheBloke/airoboros-13B-HF": ModelType.IFT, | |
"TheBloke/koala-13B-HF": ModelType.FT, | |
"TheBloke/Wizard-Vicuna-7B-Uncensored-HF": ModelType.FT, | |
"TheBloke/dromedary-65b-lora-HF": ModelType.IFT, | |
"TheBloke/wizardLM-13B-1.0-fp16": ModelType.IFT, | |
"TheBloke/WizardLM-13B-V1-1-SuperHOT-8K-fp16": ModelType.FT, | |
"TheBloke/Wizard-Vicuna-30B-Uncensored-fp16": ModelType.FT, | |
"TheBloke/wizard-vicuna-13B-HF": ModelType.IFT, | |
"TheBloke/UltraLM-13B-fp16": ModelType.IFT, | |
"TheBloke/OpenAssistant-FT-7-Llama-30B-HF": ModelType.FT, | |
"TheBloke/vicuna-13B-1.1-HF": ModelType.IFT, | |
"TheBloke/guanaco-13B-HF": ModelType.FT, | |
"TheBloke/guanaco-65B-HF": ModelType.FT, | |
"TheBloke/airoboros-7b-gpt4-fp16": ModelType.IFT, | |
"TheBloke/llama-30b-supercot-SuperHOT-8K-fp16": ModelType.IFT, | |
"TheBloke/Llama-2-13B-fp16": ModelType.PT, | |
"TheBloke/llama-2-70b-Guanaco-QLoRA-fp16": ModelType.FT, | |
"TheBloke/landmark-attention-llama7b-fp16": ModelType.IFT, | |
"TheBloke/Planner-7B-fp16": ModelType.IFT, | |
"TheBloke/Wizard-Vicuna-13B-Uncensored-HF": ModelType.FT, | |
"TheBloke/gpt4-alpaca-lora-13B-HF": ModelType.IFT, | |
"TheBloke/gpt4-x-vicuna-13B-HF": ModelType.IFT, | |
"TheBloke/gpt4-alpaca-lora_mlp-65B-HF": ModelType.IFT, | |
"TheBloke/tulu-13B-fp16": ModelType.IFT, | |
"TheBloke/VicUnlocked-alpaca-65B-QLoRA-fp16": ModelType.IFT, | |
"TheBloke/Llama-2-70B-fp16": ModelType.IFT, | |
"TheBloke/WizardLM-30B-fp16": ModelType.IFT, | |
"TheBloke/robin-13B-v2-fp16": ModelType.FT, | |
"TheBloke/robin-33B-v2-fp16": ModelType.FT, | |
"TheBloke/Vicuna-13B-CoT-fp16": ModelType.IFT, | |
"TheBloke/Vicuna-33B-1-3-SuperHOT-8K-fp16": ModelType.IFT, | |
"TheBloke/Wizard-Vicuna-30B-Superhot-8K-fp16": ModelType.FT, | |
"TheBloke/Nous-Hermes-13B-SuperHOT-8K-fp16": ModelType.IFT, | |
"TheBloke/GPlatty-30B-SuperHOT-8K-fp16": ModelType.FT, | |
"TheBloke/CAMEL-33B-Combined-Data-SuperHOT-8K-fp16": ModelType.IFT, | |
"TheBloke/Chinese-Alpaca-33B-SuperHOT-8K-fp16": ModelType.IFT, | |
"jphme/orca_mini_v2_ger_7b": ModelType.IFT, | |
"Ejafa/vicuna_7B_vanilla_1.1": ModelType.FT, | |
"kevinpro/Vicuna-13B-CoT": ModelType.IFT, | |
"AlekseyKorshuk/pygmalion-6b-vicuna-chatml": ModelType.FT, | |
"AlekseyKorshuk/chatml-pyg-v1": ModelType.FT, | |
"concedo/Vicuzard-30B-Uncensored": ModelType.FT, | |
"concedo/OPT-19M-ChatSalad": ModelType.FT, | |
"concedo/Pythia-70M-ChatSalad": ModelType.FT, | |
"digitous/13B-HyperMantis": ModelType.IFT, | |
"digitous/Adventien-GPTJ": ModelType.FT, | |
"digitous/Alpacino13b": ModelType.IFT, | |
"digitous/GPT-R": ModelType.IFT, | |
"digitous/Javelin-R": ModelType.IFT, | |
"digitous/Javalion-GPTJ": ModelType.IFT, | |
"digitous/Javalion-R": ModelType.IFT, | |
"digitous/Skegma-GPTJ": ModelType.FT, | |
"digitous/Alpacino30b": ModelType.IFT, | |
"digitous/Janin-GPTJ": ModelType.FT, | |
"digitous/Janin-R": ModelType.FT, | |
"digitous/Javelin-GPTJ": ModelType.FT, | |
"SaylorTwift/gpt2_test": ModelType.PT, | |
"anton-l/gpt-j-tiny-random": ModelType.FT, | |
"Andron00e/YetAnother_Open-Llama-3B-LoRA-OpenOrca": ModelType.FT, | |
"Lazycuber/pyg-instruct-wizardlm": ModelType.FT, | |
"Lazycuber/Janemalion-6B": ModelType.FT, | |
"IDEA-CCNL/Ziya-LLaMA-13B-Pretrain-v1": ModelType.FT, | |
"IDEA-CCNL/Ziya-LLaMA-13B-v1": ModelType.IFT, | |
"dsvv-cair/alpaca-cleaned-llama-30b-bf16": ModelType.FT, | |
"gpt2-medium": ModelType.PT, | |
"camel-ai/CAMEL-13B-Combined-Data": ModelType.IFT, | |
"camel-ai/CAMEL-13B-Role-Playing-Data": ModelType.FT, | |
"camel-ai/CAMEL-33B-Combined-Data": ModelType.IFT, | |
"PygmalionAI/pygmalion-6b": ModelType.FT, | |
"PygmalionAI/metharme-1.3b": ModelType.IFT, | |
"PygmalionAI/pygmalion-1.3b": ModelType.FT, | |
"PygmalionAI/pygmalion-350m": ModelType.FT, | |
"PygmalionAI/pygmalion-2.7b": ModelType.FT, | |
"medalpaca/medalpaca-7b": ModelType.FT, | |
"lilloukas/Platypus-30B": ModelType.IFT, | |
"lilloukas/GPlatty-30B": ModelType.FT, | |
"mncai/chatdoctor": ModelType.FT, | |
"chaoyi-wu/MedLLaMA_13B": ModelType.FT, | |
"LoupGarou/WizardCoder-Guanaco-15B-V1.0": ModelType.IFT, | |
"LoupGarou/WizardCoder-Guanaco-15B-V1.1": ModelType.FT, | |
"hakurei/instruct-12b": ModelType.IFT, | |
"hakurei/lotus-12B": ModelType.FT, | |
"shibing624/chinese-llama-plus-13b-hf": ModelType.IFT, | |
"shibing624/chinese-alpaca-plus-7b-hf": ModelType.IFT, | |
"shibing624/chinese-alpaca-plus-13b-hf": ModelType.IFT, | |
"mosaicml/mpt-7b-instruct": ModelType.IFT, | |
"mosaicml/mpt-30b-chat": ModelType.IFT, | |
"mosaicml/mpt-7b-storywriter": ModelType.FT, | |
"mosaicml/mpt-30b-instruct": ModelType.IFT, | |
"mosaicml/mpt-7b-chat": ModelType.IFT, | |
"mosaicml/mpt-30b": ModelType.PT, | |
"Corianas/111m": ModelType.IFT, | |
"Corianas/Quokka_1.3b": ModelType.IFT, | |
"Corianas/256_5epoch": ModelType.FT, | |
"Corianas/Quokka_256m": ModelType.IFT, | |
"Corianas/Quokka_590m": ModelType.IFT, | |
"Corianas/gpt-j-6B-Dolly": ModelType.FT, | |
"Corianas/Quokka_2.7b": ModelType.IFT, | |
"cyberagent/open-calm-7b": ModelType.FT, | |
"Aspik101/Nous-Hermes-13b-pl-lora_unload": ModelType.IFT, | |
"THUDM/chatglm2-6b": ModelType.IFT, | |
"MetaIX/GPT4-X-Alpasta-30b": ModelType.IFT, | |
"NYTK/PULI-GPTrio": ModelType.PT, | |
"EleutherAI/pythia-1.3b": ModelType.PT, | |
"EleutherAI/pythia-2.8b-deduped": ModelType.PT, | |
"EleutherAI/gpt-neo-125m": ModelType.PT, | |
"EleutherAI/pythia-160m": ModelType.PT, | |
"EleutherAI/gpt-neo-2.7B": ModelType.PT, | |
"EleutherAI/pythia-1b-deduped": ModelType.PT, | |
"EleutherAI/pythia-6.7b": ModelType.PT, | |
"EleutherAI/pythia-70m-deduped": ModelType.PT, | |
"EleutherAI/gpt-neox-20b": ModelType.PT, | |
"EleutherAI/pythia-1.4b-deduped": ModelType.PT, | |
"EleutherAI/pythia-2.7b": ModelType.PT, | |
"EleutherAI/pythia-6.9b-deduped": ModelType.PT, | |
"EleutherAI/pythia-70m": ModelType.PT, | |
"EleutherAI/gpt-j-6b": ModelType.PT, | |
"EleutherAI/pythia-12b-deduped": ModelType.PT, | |
"EleutherAI/gpt-neo-1.3B": ModelType.PT, | |
"EleutherAI/pythia-410m-deduped": ModelType.PT, | |
"EleutherAI/pythia-160m-deduped": ModelType.PT, | |
"EleutherAI/polyglot-ko-12.8b": ModelType.PT, | |
"EleutherAI/pythia-12b": ModelType.PT, | |
"roneneldan/TinyStories-33M": ModelType.PT, | |
"roneneldan/TinyStories-28M": ModelType.PT, | |
"roneneldan/TinyStories-1M": ModelType.PT, | |
"roneneldan/TinyStories-8M": ModelType.PT, | |
"roneneldan/TinyStories-3M": ModelType.PT, | |
"jerryjalapeno/nart-100k-7b": ModelType.FT, | |
"lmsys/vicuna-13b-v1.3": ModelType.IFT, | |
"lmsys/vicuna-7b-v1.3": ModelType.IFT, | |
"lmsys/vicuna-13b-v1.1": ModelType.IFT, | |
"lmsys/vicuna-13b-delta-v1.1": ModelType.IFT, | |
"lmsys/vicuna-7b-delta-v1.1": ModelType.IFT, | |
"abhiramtirumala/DialoGPT-sarcastic-medium": ModelType.FT, | |
"haonan-li/bactrian-x-llama-13b-merged": ModelType.IFT, | |
"Gryphe/MythoLogic-13b": ModelType.IFT, | |
"Gryphe/MythoBoros-13b": ModelType.IFT, | |
"pillowtalks-ai/delta13b": ModelType.FT, | |
"wannaphong/openthaigpt-0.1.0-beta-full-model_for_open_llm_leaderboard": ModelType.FT, | |
"bigscience/bloom-7b1": ModelType.PT, | |
"bigcode/tiny_starcoder_py": ModelType.PT, | |
"bigcode/starcoderplus": ModelType.FT, | |
"bigcode/gpt_bigcode-santacoder": ModelType.PT, | |
"bigcode/starcoder": ModelType.PT, | |
"Open-Orca/OpenOrca-Preview1-13B": ModelType.IFT, | |
"microsoft/DialoGPT-large": ModelType.FT, | |
"microsoft/DialoGPT-small": ModelType.FT, | |
"microsoft/DialoGPT-medium": ModelType.FT, | |
"microsoft/CodeGPT-small-py": ModelType.FT, | |
"Tincando/fiction_story_generator": ModelType.FT, | |
"Pirr/pythia-13b-deduped-green_devil": ModelType.FT, | |
"Aeala/GPT4-x-AlpacaDente2-30b": ModelType.FT, | |
"Aeala/GPT4-x-AlpacaDente-30b": ModelType.FT, | |
"Aeala/GPT4-x-Alpasta-13b": ModelType.FT, | |
"Aeala/VicUnlocked-alpaca-30b": ModelType.IFT, | |
"Tap-M/Luna-AI-Llama2-Uncensored": ModelType.FT, | |
"illuin/test-custom-llama": ModelType.FT, | |
"dvruette/oasst-llama-13b-2-epochs": ModelType.FT, | |
"dvruette/oasst-gpt-neox-20b-1000-steps": ModelType.FT, | |
"dvruette/llama-13b-pretrained-dropout": ModelType.PT, | |
"dvruette/llama-13b-pretrained": ModelType.PT, | |
"dvruette/llama-13b-pretrained-sft-epoch-1": ModelType.FT, | |
"dvruette/llama-13b-pretrained-sft-do2": ModelType.FT, | |
"dvruette/oasst-gpt-neox-20b-3000-steps": ModelType.FT, | |
"dvruette/oasst-pythia-12b-pretrained-sft": ModelType.FT, | |
"dvruette/oasst-pythia-6.9b-4000-steps": ModelType.FT, | |
"dvruette/gpt-neox-20b-full-precision": ModelType.FT, | |
"dvruette/oasst-llama-13b-1000-steps": ModelType.FT, | |
"openlm-research/open_llama_7b_700bt_preview": ModelType.PT, | |
"openlm-research/open_llama_7b": ModelType.PT, | |
"openlm-research/open_llama_7b_v2": ModelType.PT, | |
"openlm-research/open_llama_3b": ModelType.PT, | |
"openlm-research/open_llama_13b": ModelType.PT, | |
"openlm-research/open_llama_3b_v2": ModelType.PT, | |
"PocketDoc/Dans-PileOfSets-Mk1-llama-13b-merged": ModelType.IFT, | |
"GeorgiaTechResearchInstitute/galpaca-30b": ModelType.IFT, | |
"GeorgiaTechResearchInstitute/starcoder-gpteacher-code-instruct": ModelType.IFT, | |
"databricks/dolly-v2-7b": ModelType.IFT, | |
"databricks/dolly-v2-3b": ModelType.IFT, | |
"databricks/dolly-v2-12b": ModelType.IFT, | |
"Rachneet/gpt2-xl-alpaca": ModelType.FT, | |
"Locutusque/gpt2-conversational-or-qa": ModelType.FT, | |
"psyche/kogpt": ModelType.FT, | |
"NbAiLab/nb-gpt-j-6B-alpaca": ModelType.IFT, | |
"Mikael110/llama-2-7b-guanaco-fp16": ModelType.FT, | |
"Mikael110/llama-2-13b-guanaco-fp16": ModelType.FT, | |
"Fredithefish/CrimsonPajama": ModelType.IFT, | |
"Fredithefish/RedPajama-INCITE-Chat-3B-ShareGPT-11K": ModelType.FT, | |
"Fredithefish/ScarletPajama-3B-HF": ModelType.FT, | |
"Fredithefish/RedPajama-INCITE-Chat-3B-Instruction-Tuning-with-GPT-4": ModelType.IFT, | |
"acrastt/RedPajama-INCITE-Chat-Instruct-3B-V1": ModelType.IFT, | |
"eachadea/vicuna-13b-1.1": ModelType.FT, | |
"eachadea/vicuna-7b-1.1": ModelType.FT, | |
"eachadea/vicuna-13b": ModelType.FT, | |
"openaccess-ai-collective/wizard-mega-13b": ModelType.IFT, | |
"openaccess-ai-collective/manticore-13b": ModelType.IFT, | |
"openaccess-ai-collective/manticore-30b-chat-pyg-alpha": ModelType.IFT, | |
"openaccess-ai-collective/minotaur-13b": ModelType.IFT, | |
"openaccess-ai-collective/minotaur-13b-fixed": ModelType.IFT, | |
"openaccess-ai-collective/hippogriff-30b-chat": ModelType.IFT, | |
"openaccess-ai-collective/manticore-13b-chat-pyg": ModelType.IFT, | |
"pythainlp/wangchanglm-7.5B-sft-enth": ModelType.IFT, | |
"pythainlp/wangchanglm-7.5B-sft-en-sharded": ModelType.IFT, | |
"euclaise/gpt-neox-122m-minipile-digits": ModelType.FT, | |
"stabilityai/StableBeluga1-Delta": ModelType.IFT, | |
"stabilityai/stablelm-tuned-alpha-7b": ModelType.IFT, | |
"stabilityai/StableBeluga2": ModelType.IFT, | |
"stabilityai/StableBeluga-13B": ModelType.IFT, | |
"stabilityai/StableBeluga-7B": ModelType.IFT, | |
"stabilityai/stablelm-base-alpha-7b": ModelType.PT, | |
"stabilityai/stablelm-base-alpha-3b": ModelType.PT, | |
"stabilityai/stablelm-tuned-alpha-3b": ModelType.IFT, | |
"alibidaran/medical_transcription_generator": ModelType.FT, | |
"CalderaAI/30B-Lazarus": ModelType.IFT, | |
"CalderaAI/13B-BlueMethod": ModelType.IFT, | |
"CalderaAI/13B-Ouroboros": ModelType.IFT, | |
"KoboldAI/OPT-13B-Erebus": ModelType.FT, | |
"KoboldAI/GPT-J-6B-Janeway": ModelType.FT, | |
"KoboldAI/GPT-J-6B-Shinen": ModelType.FT, | |
"KoboldAI/fairseq-dense-2.7B": ModelType.PT, | |
"KoboldAI/OPT-6B-nerys-v2": ModelType.FT, | |
"KoboldAI/GPT-NeoX-20B-Skein": ModelType.FT, | |
"KoboldAI/PPO_Pygway-6b-Mix": ModelType.FT, | |
"KoboldAI/fairseq-dense-6.7B": ModelType.PT, | |
"KoboldAI/fairseq-dense-125M": ModelType.PT, | |
"KoboldAI/OPT-13B-Nerybus-Mix": ModelType.FT, | |
"KoboldAI/OPT-2.7B-Erebus": ModelType.FT, | |
"KoboldAI/OPT-350M-Nerys-v2": ModelType.FT, | |
"KoboldAI/OPT-2.7B-Nerys-v2": ModelType.FT, | |
"KoboldAI/OPT-2.7B-Nerybus-Mix": ModelType.FT, | |
"KoboldAI/OPT-13B-Nerys-v2": ModelType.FT, | |
"KoboldAI/GPT-NeoX-20B-Erebus": ModelType.FT, | |
"KoboldAI/OPT-6.7B-Erebus": ModelType.FT, | |
"KoboldAI/fairseq-dense-355M": ModelType.PT, | |
"KoboldAI/OPT-6.7B-Nerybus-Mix": ModelType.FT, | |
"KoboldAI/GPT-J-6B-Adventure": ModelType.FT, | |
"KoboldAI/OPT-350M-Erebus": ModelType.FT, | |
"KoboldAI/GPT-J-6B-Skein": ModelType.FT, | |
"KoboldAI/OPT-30B-Erebus": ModelType.FT, | |
"klosax/pythia-160m-deduped-step92k-193bt": ModelType.PT, | |
"klosax/open_llama_3b_350bt_preview": ModelType.PT, | |
"klosax/openllama-3b-350bt": ModelType.PT, | |
"klosax/pythia-70m-deduped-step44k-92bt": ModelType.PT, | |
"klosax/open_llama_13b_600bt_preview": ModelType.PT, | |
"klosax/open_llama_7b_400bt_preview": ModelType.PT, | |
"kfkas/Llama-2-ko-7b-Chat": ModelType.IFT, | |
"WeOpenML/Alpaca-7B-v1": ModelType.IFT, | |
"WeOpenML/PandaLM-Alpaca-7B-v1": ModelType.IFT, | |
"TFLai/gpt2-turkish-uncased": ModelType.FT, | |
"ehartford/WizardLM-13B-Uncensored": ModelType.IFT, | |
"ehartford/dolphin-llama-13b": ModelType.IFT, | |
"ehartford/Wizard-Vicuna-30B-Uncensored": ModelType.FT, | |
"ehartford/WizardLM-30B-Uncensored": ModelType.IFT, | |
"ehartford/Wizard-Vicuna-13B-Uncensored": ModelType.FT, | |
"ehartford/WizardLM-7B-Uncensored": ModelType.IFT, | |
"ehartford/based-30b": ModelType.FT, | |
"ehartford/Wizard-Vicuna-7B-Uncensored": ModelType.FT, | |
"wahaha1987/llama_7b_sharegpt94k_fastchat": ModelType.FT, | |
"wahaha1987/llama_13b_sharegpt94k_fastchat": ModelType.FT, | |
"OpenAssistant/oasst-sft-1-pythia-12b": ModelType.FT, | |
"OpenAssistant/stablelm-7b-sft-v7-epoch-3": ModelType.IFT, | |
"OpenAssistant/oasst-sft-4-pythia-12b-epoch-3.5": ModelType.FT, | |
"OpenAssistant/pythia-12b-sft-v8-2.5k-steps": ModelType.IFT, | |
"OpenAssistant/pythia-12b-sft-v8-7k-steps": ModelType.IFT, | |
"OpenAssistant/pythia-12b-pre-v8-12.5k-steps": ModelType.IFT, | |
"OpenAssistant/llama2-13b-orca-8k-3319": ModelType.IFT, | |
"junelee/wizard-vicuna-13b": ModelType.FT, | |
"BreadAi/gpt-YA-1-1_160M": ModelType.PT, | |
"BreadAi/MuseCan": ModelType.PT, | |
"BreadAi/MusePy-1-2": ModelType.PT, | |
"BreadAi/DiscordPy": ModelType.PT, | |
"BreadAi/PM_modelV2": ModelType.PT, | |
"BreadAi/gpt-Youtube": ModelType.PT, | |
"BreadAi/StoryPy": ModelType.FT, | |
"julianweng/Llama-2-7b-chat-orcah": ModelType.FT, | |
"AGI-inc/lora_moe_7b_baseline": ModelType.FT, | |
"AGI-inc/lora_moe_7b": ModelType.FT, | |
"togethercomputer/GPT-NeoXT-Chat-Base-20B": ModelType.IFT, | |
"togethercomputer/RedPajama-INCITE-Chat-7B-v0.1": ModelType.IFT, | |
"togethercomputer/RedPajama-INCITE-Instruct-7B-v0.1": ModelType.IFT, | |
"togethercomputer/RedPajama-INCITE-7B-Base": ModelType.PT, | |
"togethercomputer/RedPajama-INCITE-7B-Instruct": ModelType.IFT, | |
"togethercomputer/RedPajama-INCITE-Base-3B-v1": ModelType.PT, | |
"togethercomputer/Pythia-Chat-Base-7B": ModelType.IFT, | |
"togethercomputer/RedPajama-INCITE-Base-7B-v0.1": ModelType.PT, | |
"togethercomputer/GPT-JT-6B-v1": ModelType.IFT, | |
"togethercomputer/GPT-JT-6B-v0": ModelType.IFT, | |
"togethercomputer/RedPajama-INCITE-Chat-3B-v1": ModelType.IFT, | |
"togethercomputer/RedPajama-INCITE-7B-Chat": ModelType.IFT, | |
"togethercomputer/RedPajama-INCITE-Instruct-3B-v1": ModelType.IFT, | |
"Writer/camel-5b-hf": ModelType.IFT, | |
"Writer/palmyra-base": ModelType.PT, | |
"MBZUAI/LaMini-GPT-1.5B": ModelType.IFT, | |
"MBZUAI/lamini-cerebras-111m": ModelType.IFT, | |
"MBZUAI/lamini-neo-1.3b": ModelType.IFT, | |
"MBZUAI/lamini-cerebras-1.3b": ModelType.IFT, | |
"MBZUAI/lamini-cerebras-256m": ModelType.IFT, | |
"MBZUAI/LaMini-GPT-124M": ModelType.IFT, | |
"MBZUAI/lamini-neo-125m": ModelType.IFT, | |
"TehVenom/DiffMerge-DollyGPT-Pygmalion": ModelType.FT, | |
"TehVenom/PPO_Shygmalion-6b": ModelType.FT, | |
"TehVenom/Dolly_Shygmalion-6b-Dev_V8P2": ModelType.FT, | |
"TehVenom/Pygmalion_AlpacaLora-7b": ModelType.FT, | |
"TehVenom/PPO_Pygway-V8p4_Dev-6b": ModelType.FT, | |
"TehVenom/Dolly_Malion-6b": ModelType.FT, | |
"TehVenom/PPO_Shygmalion-V8p4_Dev-6b": ModelType.FT, | |
"TehVenom/ChanMalion": ModelType.FT, | |
"TehVenom/GPT-J-Pyg_PPO-6B": ModelType.IFT, | |
"TehVenom/Pygmalion-13b-Merged": ModelType.FT, | |
"TehVenom/Metharme-13b-Merged": ModelType.IFT, | |
"TehVenom/Dolly_Shygmalion-6b": ModelType.FT, | |
"TehVenom/GPT-J-Pyg_PPO-6B-Dev-V8p4": ModelType.IFT, | |
"georgesung/llama2_7b_chat_uncensored": ModelType.FT, | |
"vicgalle/gpt2-alpaca": ModelType.IFT, | |
"vicgalle/alpaca-7b": ModelType.FT, | |
"vicgalle/gpt2-alpaca-gpt4": ModelType.IFT, | |
"facebook/opt-350m": ModelType.PT, | |
"facebook/opt-125m": ModelType.PT, | |
"facebook/xglm-4.5B": ModelType.PT, | |
"facebook/opt-2.7b": ModelType.PT, | |
"facebook/opt-6.7b": ModelType.PT, | |
"facebook/galactica-30b": ModelType.PT, | |
"facebook/opt-13b": ModelType.PT, | |
"facebook/opt-66b": ModelType.PT, | |
"facebook/xglm-7.5B": ModelType.PT, | |
"facebook/xglm-564M": ModelType.PT, | |
"facebook/opt-30b": ModelType.PT, | |
"golaxy/gogpt-7b": ModelType.FT, | |
"golaxy/gogpt2-7b": ModelType.FT, | |
"golaxy/gogpt-7b-bloom": ModelType.FT, | |
"golaxy/gogpt-3b-bloom": ModelType.FT, | |
"psmathur/orca_mini_v2_7b": ModelType.IFT, | |
"psmathur/orca_mini_7b": ModelType.IFT, | |
"psmathur/orca_mini_3b": ModelType.IFT, | |
"psmathur/orca_mini_v2_13b": ModelType.IFT, | |
"gpt2-xl": ModelType.PT, | |
"lxe/Cerebras-GPT-2.7B-Alpaca-SP": ModelType.FT, | |
"Monero/Manticore-13b-Chat-Pyg-Guanaco": ModelType.FT, | |
"Monero/WizardLM-Uncensored-SuperCOT-StoryTelling-30b": ModelType.IFT, | |
"Monero/WizardLM-13b-OpenAssistant-Uncensored": ModelType.IFT, | |
"Monero/WizardLM-30B-Uncensored-Guanaco-SuperCOT-30b": ModelType.IFT, | |
"jzjiao/opt-1.3b-rlhf": ModelType.FT, | |
"HuggingFaceH4/starchat-beta": ModelType.IFT, | |
"KnutJaegersberg/gpt-2-xl-EvolInstruct": ModelType.IFT, | |
"KnutJaegersberg/megatron-GPT-2-345m-EvolInstruct": ModelType.IFT, | |
"KnutJaegersberg/galactica-orca-wizardlm-1.3b": ModelType.IFT, | |
"openchat/openchat_8192": ModelType.IFT, | |
"openchat/openchat_v2": ModelType.IFT, | |
"openchat/openchat_v2_w": ModelType.IFT, | |
"ausboss/llama-13b-supercot": ModelType.IFT, | |
"ausboss/llama-30b-supercot": ModelType.IFT, | |
"Neko-Institute-of-Science/metharme-7b": ModelType.IFT, | |
"Neko-Institute-of-Science/pygmalion-7b": ModelType.FT, | |
"SebastianSchramm/Cerebras-GPT-111M-instruction": ModelType.IFT, | |
"victor123/WizardLM-13B-1.0": ModelType.IFT, | |
"OpenBuddy/openbuddy-openllama-13b-v7-fp16": ModelType.FT, | |
"OpenBuddy/openbuddy-llama2-13b-v8.1-fp16": ModelType.FT, | |
"OpenBuddyEA/openbuddy-llama-30b-v7.1-bf16": ModelType.FT, | |
"baichuan-inc/Baichuan-7B": ModelType.PT, | |
"tiiuae/falcon-40b-instruct": ModelType.IFT, | |
"tiiuae/falcon-40b": ModelType.PT, | |
"tiiuae/falcon-7b": ModelType.PT, | |
"YeungNLP/firefly-llama-13b": ModelType.FT, | |
"YeungNLP/firefly-llama-13b-v1.2": ModelType.FT, | |
"YeungNLP/firefly-llama2-13b": ModelType.FT, | |
"YeungNLP/firefly-ziya-13b": ModelType.FT, | |
"shaohang/Sparse0.5_OPT-1.3": ModelType.FT, | |
"xzuyn/Alpacino-SuperCOT-13B": ModelType.IFT, | |
"xzuyn/MedicWizard-7B": ModelType.FT, | |
"xDAN-AI/xDAN_13b_l2_lora": ModelType.FT, | |
"beomi/KoAlpaca-Polyglot-5.8B": ModelType.FT, | |
"beomi/llama-2-ko-7b": ModelType.IFT, | |
"Salesforce/codegen-6B-multi": ModelType.PT, | |
"Salesforce/codegen-16B-nl": ModelType.PT, | |
"Salesforce/codegen-6B-nl": ModelType.PT, | |
"ai-forever/rugpt3large_based_on_gpt2": ModelType.FT, | |
"gpt2-large": ModelType.PT, | |
"frank098/orca_mini_3b_juniper": ModelType.FT, | |
"frank098/WizardLM_13B_juniper": ModelType.FT, | |
"FPHam/Free_Sydney_13b_HF": ModelType.FT, | |
"huggingface/llama-13b": ModelType.PT, | |
"huggingface/llama-7b": ModelType.PT, | |
"huggingface/llama-65b": ModelType.PT, | |
"huggingface/llama-30b": ModelType.PT, | |
"Henk717/chronoboros-33B": ModelType.IFT, | |
"jondurbin/airoboros-13b-gpt4-1.4": ModelType.IFT, | |
"jondurbin/airoboros-7b": ModelType.IFT, | |
"jondurbin/airoboros-7b-gpt4": ModelType.IFT, | |
"jondurbin/airoboros-7b-gpt4-1.1": ModelType.IFT, | |
"jondurbin/airoboros-7b-gpt4-1.2": ModelType.IFT, | |
"jondurbin/airoboros-7b-gpt4-1.3": ModelType.IFT, | |
"jondurbin/airoboros-7b-gpt4-1.4": ModelType.IFT, | |
"jondurbin/airoboros-l2-7b-gpt4-1.4.1": ModelType.IFT, | |
"jondurbin/airoboros-l2-13b-gpt4-1.4.1": ModelType.IFT, | |
"jondurbin/airoboros-l2-70b-gpt4-1.4.1": ModelType.IFT, | |
"jondurbin/airoboros-13b": ModelType.IFT, | |
"jondurbin/airoboros-33b-gpt4-1.4": ModelType.IFT, | |
"jondurbin/airoboros-33b-gpt4-1.2": ModelType.IFT, | |
"jondurbin/airoboros-65b-gpt4-1.2": ModelType.IFT, | |
"ariellee/SuperPlatty-30B": ModelType.IFT, | |
"danielhanchen/open_llama_3b_600bt_preview": ModelType.FT, | |
"cerebras/Cerebras-GPT-256M": ModelType.PT, | |
"cerebras/Cerebras-GPT-1.3B": ModelType.PT, | |
"cerebras/Cerebras-GPT-13B": ModelType.PT, | |
"cerebras/Cerebras-GPT-2.7B": ModelType.PT, | |
"cerebras/Cerebras-GPT-111M": ModelType.PT, | |
"cerebras/Cerebras-GPT-6.7B": ModelType.PT, | |
"Yhyu13/oasst-rlhf-2-llama-30b-7k-steps-hf": ModelType.RL, | |
"Yhyu13/llama-30B-hf-openassitant": ModelType.FT, | |
"NousResearch/Nous-Hermes-Llama2-13b": ModelType.IFT, | |
"NousResearch/Nous-Hermes-llama-2-7b": ModelType.IFT, | |
"NousResearch/Redmond-Puffin-13B": ModelType.IFT, | |
"NousResearch/Nous-Hermes-13b": ModelType.IFT, | |
"project-baize/baize-v2-7b": ModelType.IFT, | |
"project-baize/baize-v2-13b": ModelType.IFT, | |
"LLMs/WizardLM-13B-V1.0": ModelType.FT, | |
"LLMs/AlpacaGPT4-7B-elina": ModelType.FT, | |
"wenge-research/yayi-7b": ModelType.FT, | |
"wenge-research/yayi-7b-llama2": ModelType.FT, | |
"wenge-research/yayi-13b-llama2": ModelType.FT, | |
"yhyhy3/open_llama_7b_v2_med_instruct": ModelType.IFT, | |
"llama-anon/instruct-13b": ModelType.IFT, | |
"huggingtweets/jerma985": ModelType.FT, | |
"huggingtweets/gladosystem": ModelType.FT, | |
"huggingtweets/bladeecity-jerma985": ModelType.FT, | |
"huggyllama/llama-13b": ModelType.PT, | |
"huggyllama/llama-65b": ModelType.PT, | |
"FabbriSimo01/Facebook_opt_1.3b_Quantized": ModelType.PT, | |
"upstage/Llama-2-70b-instruct": ModelType.IFT, | |
"upstage/Llama-2-70b-instruct-1024": ModelType.IFT, | |
"upstage/llama-65b-instruct": ModelType.IFT, | |
"upstage/llama-30b-instruct-2048": ModelType.IFT, | |
"upstage/llama-30b-instruct": ModelType.IFT, | |
"WizardLM/WizardLM-13B-1.0": ModelType.IFT, | |
"WizardLM/WizardLM-13B-V1.1": ModelType.IFT, | |
"WizardLM/WizardLM-13B-V1.2": ModelType.IFT, | |
"WizardLM/WizardLM-30B-V1.0": ModelType.IFT, | |
"WizardLM/WizardCoder-15B-V1.0": ModelType.IFT, | |
"gpt2": ModelType.PT, | |
"keyfan/vicuna-chinese-replication-v1.1": ModelType.IFT, | |
"nthngdy/pythia-owt2-70m-100k": ModelType.FT, | |
"nthngdy/pythia-owt2-70m-50k": ModelType.FT, | |
"quantumaikr/KoreanLM-hf": ModelType.FT, | |
"quantumaikr/open_llama_7b_hf": ModelType.FT, | |
"quantumaikr/QuantumLM-70B-hf": ModelType.IFT, | |
"MayaPH/FinOPT-Lincoln": ModelType.FT, | |
"MayaPH/FinOPT-Franklin": ModelType.FT, | |
"MayaPH/GodziLLa-30B": ModelType.IFT, | |
"MayaPH/GodziLLa-30B-plus": ModelType.IFT, | |
"MayaPH/FinOPT-Washington": ModelType.FT, | |
"ogimgio/gpt-neo-125m-neurallinguisticpioneers": ModelType.FT, | |
"layoric/llama-2-13b-code-alpaca": ModelType.FT, | |
"CobraMamba/mamba-gpt-3b": ModelType.FT, | |
"CobraMamba/mamba-gpt-3b-v2": ModelType.FT, | |
"CobraMamba/mamba-gpt-3b-v3": ModelType.FT, | |
"timdettmers/guanaco-33b-merged": ModelType.FT, | |
"elinas/chronos-33b": ModelType.IFT, | |
"heegyu/RedTulu-Uncensored-3B-0719": ModelType.IFT, | |
"heegyu/WizardVicuna-Uncensored-3B-0719": ModelType.IFT, | |
"heegyu/WizardVicuna-3B-0719": ModelType.IFT, | |
"meta-llama/Llama-2-7b-chat-hf": ModelType.RL, | |
"meta-llama/Llama-2-7b-hf": ModelType.PT, | |
"meta-llama/Llama-2-13b-chat-hf": ModelType.RL, | |
"meta-llama/Llama-2-13b-hf": ModelType.PT, | |
"meta-llama/Llama-2-70b-chat-hf": ModelType.RL, | |
"meta-llama/Llama-2-70b-hf": ModelType.PT, | |
"xhyi/PT_GPTNEO350_ATG": ModelType.FT, | |
"h2oai/h2ogpt-gm-oasst1-en-1024-20b": ModelType.FT, | |
"h2oai/h2ogpt-gm-oasst1-en-1024-open-llama-7b-preview-400bt": ModelType.FT, | |
"h2oai/h2ogpt-oig-oasst1-512-6_9b": ModelType.IFT, | |
"h2oai/h2ogpt-oasst1-512-12b": ModelType.IFT, | |
"h2oai/h2ogpt-oig-oasst1-256-6_9b": ModelType.IFT, | |
"h2oai/h2ogpt-gm-oasst1-en-2048-open-llama-7b-preview-300bt": ModelType.FT, | |
"h2oai/h2ogpt-oasst1-512-20b": ModelType.IFT, | |
"h2oai/h2ogpt-gm-oasst1-en-2048-open-llama-7b-preview-300bt-v2": ModelType.FT, | |
"h2oai/h2ogpt-gm-oasst1-en-1024-12b": ModelType.FT, | |
"h2oai/h2ogpt-gm-oasst1-multilang-1024-20b": ModelType.FT, | |
"bofenghuang/vigogne-13b-instruct": ModelType.IFT, | |
"bofenghuang/vigogne-13b-chat": ModelType.FT, | |
"bofenghuang/vigogne-2-7b-instruct": ModelType.IFT, | |
"bofenghuang/vigogne-7b-instruct": ModelType.IFT, | |
"bofenghuang/vigogne-7b-chat": ModelType.FT, | |
"Vmware/open-llama-7b-v2-open-instruct": ModelType.IFT, | |
"VMware/open-llama-0.7T-7B-open-instruct-v1.1": ModelType.IFT, | |
"ewof/koishi-instruct-3b": ModelType.IFT, | |
"gywy/llama2-13b-chinese-v1": ModelType.FT, | |
"GOAT-AI/GOAT-7B-Community": ModelType.FT, | |
"psyche/kollama2-7b": ModelType.FT, | |
"TheTravellingEngineer/llama2-7b-hf-guanaco": ModelType.FT, | |
"beaugogh/pythia-1.4b-deduped-sharegpt": ModelType.FT, | |
"augtoma/qCammel-70-x": ModelType.IFT, | |
"Lajonbot/Llama-2-7b-chat-hf-instruct-pl-lora_unload": ModelType.IFT, | |
"anhnv125/pygmalion-6b-roleplay": ModelType.FT, | |
"64bits/LexPodLM-13B": ModelType.FT, | |
} | |
def model_type_from_str(type): | |
if "fine-tuned" in type or "🔶" in type: | |
return ModelType.FT | |
if "pretrained" in type or "🟢" in type: | |
return ModelType.PT | |
if "RL-tuned" in type or "🟦" in type: | |
return ModelType.RL | |
if "instruction-tuned" in type or "⭕" in type: | |
return ModelType.IFT | |
return ModelType.Unknown | |