open_llm_leaderboard

Runtime error

App Files Files Community

open_llm_leaderboard / src /display_models /model_metadata_type.py

pngwn HF staff

Duplicate from HuggingFaceH4/open_llm_leaderboard

b6cfc50 10 months ago

raw history blame contribute delete

No virus

27.3 kB

	from dataclasses import dataclass
	from enum import Enum
	from typing import Dict


	@dataclass
	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