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] = { | |
| '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 | |