-
Notifications
You must be signed in to change notification settings - Fork 4.5k
Commit
This commit does not belong to any branch on this repository, and may belong to a fork outside of the repository.
feat: Update AI/ML Models list (#3623)
* feat: Update AI/ML Models list Co-authored-by: Vinícios Batista da Silva <[email protected]>" * [autofix.ci] apply automated fixes --------- Co-authored-by: autofix-ci[bot] <114827586+autofix-ci[bot]@users.noreply.github.com>
- Loading branch information
1 parent
1a5752c
commit 387ea32
Showing
1 changed file
with
222 additions
and
50 deletions.
There are no files selected for viewing
272 changes: 222 additions & 50 deletions
272
src/backend/base/langflow/base/models/aiml_constants.py
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
Original file line number | Diff line number | Diff line change |
---|---|---|
@@ -1,80 +1,252 @@ | ||
AIML_CHAT_MODELS = [ | ||
"zero-one-ai/Yi-34B-Chat", | ||
"#g1_aura-angus-en", | ||
"#g1_aura-arcas-en", | ||
"#g1_aura-asteria-en", | ||
"#g1_aura-athena-en", | ||
"#g1_aura-helios-en", | ||
"#g1_aura-hera-en", | ||
"#g1_aura-luna-en", | ||
"#g1_aura-orion-en", | ||
"#g1_aura-orpheus-en", | ||
"#g1_aura-perseus-en", | ||
"#g1_aura-stella-en", | ||
"#g1_aura-zeus-en", | ||
"#g1_nova-2-automotive", | ||
"#g1_nova-2-conversationalai", | ||
"#g1_nova-2-drivethru", | ||
"#g1_nova-2-finance", | ||
"#g1_nova-2-general", | ||
"#g1_nova-2-medical", | ||
"#g1_nova-2-meeting", | ||
"#g1_nova-2-phonecall", | ||
"#g1_nova-2-video", | ||
"#g1_nova-2-voicemail", | ||
"#g1_redaction", | ||
"#g1_whisper-base", | ||
"#g1_whisper-large", | ||
"#g1_whisper-medium", | ||
"#g1_whisper-small", | ||
"#g1_whisper-tiny", | ||
"Austism/chronos-hermes-13b", | ||
"BAAI/bge-base-en-v1.5", | ||
"BAAI/bge-large-en-v1.5", | ||
"EleutherAI/llemma_7b", | ||
"Gryphe/MythoMax-L2-13b", | ||
"HuggingFaceH4/zephyr-7b-beta", | ||
"Meta-Llama/Llama-Guard-7b", | ||
"Nexusflow/NexusRaven-V2-13B", | ||
"NousResearch/Hermes-2-Theta-Llama-3-70B", | ||
"NousResearch/Nous-Capybara-7B-V1p9", | ||
"NousResearch/Nous-Hermes-13b", | ||
"NousResearch/Nous-Hermes-2-Mistral-7B-DPO", | ||
"NousResearch/Nous-Hermes-2-Mixtral-8x7B-DPO", | ||
"NousResearch/Nous-Hermes-2-Mixtral-8x7B-SFT", | ||
"NousResearch/Nous-Hermes-2-Yi-34B", | ||
"NousResearch/Nous-Hermes-Llama2-13b", | ||
"NousResearch/Nous-Hermes-Llama2-70b", | ||
"NousResearch/Nous-Hermes-llama-2-7b", | ||
"NumbersStation/nsql-llama-2-7B", | ||
"Open-Orca/Mistral-7B-OpenOrca", | ||
"Phind/Phind-CodeLlama-34B-Python-v1", | ||
"Phind/Phind-CodeLlama-34B-v2", | ||
"Qwen/Qwen1.5-0.5B", | ||
"Qwen/Qwen1.5-0.5B-Chat", | ||
"Qwen/Qwen1.5-1.8B", | ||
"Qwen/Qwen1.5-1.8B-Chat", | ||
"Qwen/Qwen1.5-110B-Chat", | ||
"Qwen/Qwen1.5-14B", | ||
"Qwen/Qwen1.5-14B-Chat", | ||
"Qwen/Qwen1.5-32B", | ||
"Qwen/Qwen1.5-32B-Chat", | ||
"Qwen/Qwen1.5-4B", | ||
"Qwen/Qwen1.5-4B-Chat", | ||
"Qwen/Qwen1.5-72B", | ||
"Qwen/Qwen1.5-72B-Chat", | ||
"Qwen/Qwen1.5-7B", | ||
"Qwen/Qwen1.5-7B-Chat", | ||
"Qwen/Qwen2-1.5B", | ||
"Qwen/Qwen2-1.5B-Instruct", | ||
"Qwen/Qwen2-72B", | ||
"Qwen/Qwen2-72B-Instruct", | ||
"Qwen/Qwen2-7B", | ||
"Qwen/Qwen2-7B-Instruct", | ||
"SG161222/Realistic_Vision_V3.0_VAE", | ||
"Snowflake/snowflake-arctic-instruct", | ||
"Undi95/ReMM-SLERP-L2-13B", | ||
"Undi95/Toppy-M-7B", | ||
"WhereIsAI/UAE-Large-V1", | ||
"WizardLM/WizardCoder-Python-34B-V1.0", | ||
"WizardLM/WizardLM-13B-V1.2", | ||
"WizardLM/WizardLM-70B-V1.0", | ||
"allenai/OLMo-7B", | ||
"allenai/OLMo-7B-Instruct", | ||
"allenai/OLMo-7B-Twin-2T", | ||
"allenai/OLMo-7B", | ||
"Austism/chronos-hermes-13b", | ||
"bert-base-uncased", | ||
"carson/ml318br", | ||
"chatgpt-4o-latest", | ||
"claude-3-5-sonnet-20240620", | ||
"claude-3-haiku-20240307", | ||
"claude-3-opus-20240229", | ||
"claude-3-sonnet-20240229", | ||
"codellama/CodeLlama-13b-Instruct-hf", | ||
"codellama/CodeLlama-13b-Python-hf", | ||
"codellama/CodeLlama-13b-hf", | ||
"codellama/CodeLlama-34b-Instruct-hf", | ||
"codellama/CodeLlama-34b-Python-hf", | ||
"codellama/CodeLlama-34b-hf", | ||
"codellama/CodeLlama-70b-Instruct-hf", | ||
"codellama/CodeLlama-70b-Python-hf", | ||
"codellama/CodeLlama-70b-hf", | ||
"codellama/CodeLlama-7b-Instruct-hf", | ||
"codellama/CodeLlama-7b-Python-hf", | ||
"codellama/CodeLlama-7b-hf", | ||
"cognitivecomputations/dolphin-2.5-mixtral-8x7b", | ||
"dall-e-2", | ||
"dall-e-3", | ||
"databricks/dbrx-instruct", | ||
"deepseek-ai/deepseek-coder-33b-instruct", | ||
"deepseek-ai/deepseek-llm-67b-chat", | ||
"flux-pro", | ||
"flux-realism", | ||
"flux/dev", | ||
"flux/dev/image-to-image", | ||
"flux/schnell", | ||
"garage-bAInd/Platypus2-70B-instruct", | ||
"gemini-1.5-flash", | ||
"gemini-1.5-pro", | ||
"gemini-pro", | ||
"google/gemma-2-27b-it", | ||
"google/gemma-2-9b-it", | ||
"google/gemma-2b", | ||
"google/gemma-2b-it", | ||
"google/gemma-7b", | ||
"google/gemma-7b-it", | ||
"Gryphe/MythoMax-L2-13b", | ||
"gpt-3.5-turbo-0125", | ||
"gpt-3.5-turbo-0301", | ||
"gpt-3.5-turbo-0613", | ||
"gpt-3.5-turbo-1106", | ||
"gpt-3.5-turbo-16k-0613", | ||
"gpt-3.5-turbo-instruct", | ||
"gpt-4", | ||
"gpt-4-0125-preview", | ||
"gpt-4-1106-preview", | ||
"gpt-4-32k", | ||
"gpt-4-turbo", | ||
"gpt-4-turbo-2024-04-09", | ||
"gpt-4-vision-preview", | ||
"gpt-4o", | ||
"gpt-4o-2024-05-13", | ||
"gpt-4o-2024-08-06", | ||
"gpt-4o-mini", | ||
"gpt-4o-mini-2024-07-18", | ||
"gradientai/Llama-3-70B-Instruct-Gradient-1048k", | ||
"huggyllama/llama-13b", | ||
"huggyllama/llama-30b", | ||
"huggyllama/llama-65b", | ||
"huggyllama/llama-7b", | ||
"lmsys/vicuna-13b-v1.3", | ||
"lmsys/vicuna-13b-v1.5", | ||
"lmsys/vicuna-13b-v1.5-16k", | ||
"lmsys/vicuna-7b-v1.3", | ||
"lmsys/vicuna-7b-v1.5", | ||
"codellama/CodeLlama-13b-Instruct-hf", | ||
"codellama/CodeLlama-34b-Instruct-hf", | ||
"codellama/CodeLlama-70b-Instruct-hf", | ||
"codellama/CodeLlama-7b-Instruct-hf", | ||
"meta-llama/Llama-2-13b-chat-hf", | ||
"meta-llama/Llama-2-13b-hf", | ||
"meta-llama/Llama-2-70b-chat-hf", | ||
"meta-llama/Llama-2-70b-hf", | ||
"meta-llama/Llama-2-7b-chat-hf", | ||
"meta-llama/Llama-2-7b-hf", | ||
"meta-llama/Llama-3-70b-chat-hf", | ||
"meta-llama/Llama-3-70b-hf", | ||
"meta-llama/Llama-3-8b-chat-hf", | ||
"meta-llama/Llama-3-8b-hf", | ||
"meta-llama/LlamaGuard-2-8b", | ||
"meta-llama/Meta-Llama-3-70B", | ||
"meta-llama/Meta-Llama-3-70B-Instruct", | ||
"meta-llama/Meta-Llama-3-70B-Instruct-Lite", | ||
"meta-llama/Meta-Llama-3-70B-Instruct-Turbo", | ||
"meta-llama/Meta-Llama-3-8B", | ||
"meta-llama/Meta-Llama-3-8B-Instruct", | ||
"meta-llama/Meta-Llama-3-8B-Instruct-Lite", | ||
"meta-llama/Meta-Llama-3-8B-Instruct-Turbo", | ||
"meta-llama/Meta-Llama-3.1-405B-Instruct-Turbo", | ||
"meta-llama/Meta-Llama-3.1-70B-Instruct-Reference", | ||
"meta-llama/Meta-Llama-3.1-70B-Instruct-Turbo", | ||
"meta-llama/Meta-Llama-3.1-70B-Reference", | ||
"meta-llama/Meta-Llama-3.1-8B-Instruct-Turbo", | ||
"meta-llama/Llama-2-70b-chat-hf", | ||
"meta-llama/Llama-2-13b-chat-hf", | ||
"meta-llama/Llama-2-7b-chat-hf", | ||
"meta-llama/Meta-Llama-3.1-8B-Reference", | ||
"meta-llama/Meta-Llama-Guard-3-8B", | ||
"microsoft/WizardLM-2-8x22B", | ||
"microsoft/phi-2", | ||
"mistralai/Mistral-7B-Instruct-v0.1", | ||
"mistralai/Mistral-7B-Instruct-v0.2", | ||
"mistralai/Mistral-7B-Instruct-v0.3", | ||
"mistralai/Mistral-7B-v0.1", | ||
"mistralai/Mixtral-8x22B", | ||
"mistralai/Mixtral-8x22B-Instruct-v0.1", | ||
"mistralai/Mixtral-8x7B-Instruct-v0.1", | ||
"NousResearch/Nous-Capybara-7B-V1p9", | ||
"NousResearch/Nous-Hermes-2-Mixtral-8x7B-DPO", | ||
"NousResearch/Nous-Hermes-2-Mixtral-8x7B-SFT", | ||
"NousResearch/Nous-Hermes-llama-2-7b", | ||
"NousResearch/Nous-Hermes-Llama2-13b", | ||
"NousResearch/Nous-Hermes-2-Yi-34B", | ||
"mistralai/Mixtral-8x7B-v0.1", | ||
"openchat/openchat-3.5-1210", | ||
"Open-Orca/Mistral-7B-OpenOrca", | ||
"togethercomputer/Qwen-7B-Chat", | ||
"Qwen/Qwen1.5-0.5B-Chat", | ||
"Qwen/Qwen1.5-1.8B-Chat", | ||
"Qwen/Qwen1.5-4B-Chat", | ||
"Qwen/Qwen1.5-7B-Chat", | ||
"Qwen/Qwen1.5-14B-Chat", | ||
"Qwen/Qwen1.5-72B-Chat", | ||
"prompthero/openjourney", | ||
"runwayml/stable-diffusion-v1-5", | ||
"sentence-transformers/msmarco-bert-base-dot-v5", | ||
"snorkelai/Snorkel-Mistral-PairRM-DPO", | ||
"togethercomputer/alpaca-7b", | ||
"stabilityai/stable-diffusion-2-1", | ||
"stabilityai/stable-diffusion-xl-base-1.0", | ||
"stable-diffusion-v3-medium", | ||
"teknium/OpenHermes-2-Mistral-7B", | ||
"teknium/OpenHermes-2p5-Mistral-7B", | ||
"togethercomputer/falcon-40b-instruct", | ||
"togethercomputer/falcon-7b-instruct", | ||
"togethercomputer/CodeLlama-13b-Instruct", | ||
"togethercomputer/CodeLlama-13b-Python", | ||
"togethercomputer/CodeLlama-34b", | ||
"togethercomputer/CodeLlama-34b-Instruct", | ||
"togethercomputer/CodeLlama-34b-Python", | ||
"togethercomputer/CodeLlama-7b-Instruct", | ||
"togethercomputer/CodeLlama-7b-Python", | ||
"togethercomputer/Koala-13B", | ||
"togethercomputer/Koala-7B", | ||
"togethercomputer/LLaMA-2-7B-32K", | ||
"togethercomputer/Llama-2-7B-32K-Instruct", | ||
"togethercomputer/RedPajama-INCITE-Chat-3B-v1", | ||
"togethercomputer/RedPajama-INCITE-7B-Chat", | ||
"togethercomputer/Llama-3-8b-chat-hf-int4", | ||
"togethercomputer/Llama-3-8b-chat-hf-int8", | ||
"togethercomputer/SOLAR-10.7B-Instruct-v1.0-int4", | ||
"togethercomputer/StripedHyena-Hessian-7B", | ||
"togethercomputer/StripedHyena-Nous-7B", | ||
"Undi95/ReMM-SLERP-L2-13B", | ||
"Undi95/Toppy-M-7B", | ||
"WizardLM/WizardLM-13B-V1.2", | ||
"togethercomputer/alpaca-7b", | ||
"togethercomputer/evo-1-131k-base", | ||
"togethercomputer/evo-1-8k-base", | ||
"togethercomputer/guanaco-13b", | ||
"togethercomputer/guanaco-33b", | ||
"togethercomputer/guanaco-65b", | ||
"togethercomputer/guanaco-7b", | ||
"togethercomputer/llama-2-13b", | ||
"togethercomputer/llama-2-13b-chat", | ||
"togethercomputer/llama-2-70b", | ||
"togethercomputer/llama-2-70b-chat", | ||
"togethercomputer/llama-2-7b", | ||
"togethercomputer/llama-2-7b-chat", | ||
"togethercomputer/m2-bert-80M-2k-retrieval", | ||
"togethercomputer/m2-bert-80M-32k-retrieval", | ||
"togethercomputer/m2-bert-80M-8k-retrieval", | ||
"upstage/SOLAR-10.7B-Instruct-v1.0", | ||
"gpt-4", | ||
"gpt-4-turbo", | ||
"gpt-4-0613", | ||
"gpt-4-32k", | ||
"gpt-4-32k-0613", | ||
"gpt-3.5-turbo-0125", | ||
"gpt-3.5-turbo", | ||
"gpt-3.5-turbo-1106", | ||
"gpt-3.5-turbo-instruct", | ||
"gpt-3.5-turbo-16k", | ||
"gpt-3.5-turbo-0613", | ||
"gpt-3.5-turbo-16k-0613", | ||
"gpt-4o", | ||
"claude-3-opus-20240229", | ||
"claude-3-sonnet-20240229", | ||
"claude-3-haiku-20240307", | ||
"voyage-2", | ||
"voyage-code-2", | ||
"voyage-finance-2", | ||
"voyage-large-2", | ||
"voyage-large-2-instruct", | ||
"voyage-law-2", | ||
"voyage-multilingual-2", | ||
"wavymulder/Analog-Diffusion", | ||
"zero-one-ai/Yi-34B", | ||
"zero-one-ai/Yi-34B-Chat", | ||
"zero-one-ai/Yi-6B", | ||
] | ||
|
||
AIML_EMBEDDING_MODELS = [ | ||
"text-embedding-3-small", | ||
"text-embedding-3-large", | ||
"text-embedding-3-small", | ||
"text-embedding-ada-002", | ||
"text-multilingual-embedding-002", | ||
"textembedding-gecko-multilingual@001", | ||
"textembedding-gecko@001", | ||
"textembedding-gecko@003", | ||
] |