From 04838b0e211e305d8a8d334b29b2d62c71f4b6b2 Mon Sep 17 00:00:00 2001 From: Vinicios Silva Date: Thu, 29 Aug 2024 20:15:11 -0300 Subject: [PATCH 1/2] feat: Update AI/ML Models list MIME-Version: 1.0 Content-Type: text/plain; charset=UTF-8 Content-Transfer-Encoding: 8bit Co-authored-by: Vinícios Batista da Silva " --- .../langflow/base/models/aiml_constants.py | 276 ++++++++++++++---- 1 file changed, 224 insertions(+), 52 deletions(-) diff --git a/src/backend/base/langflow/base/models/aiml_constants.py b/src/backend/base/langflow/base/models/aiml_constants.py index a962190101a4..5af3b3b4c40b 100644 --- a/src/backend/base/langflow/base/models/aiml_constants.py +++ b/src/backend/base/langflow/base/models/aiml_constants.py @@ -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", + ] \ No newline at end of file From 098f832170480e687f1b0a771c220e2c2e492c66 Mon Sep 17 00:00:00 2001 From: "autofix-ci[bot]" <114827586+autofix-ci[bot]@users.noreply.github.com> Date: Thu, 29 Aug 2024 23:27:30 +0000 Subject: [PATCH 2/2] [autofix.ci] apply automated fixes --- src/backend/base/langflow/base/models/aiml_constants.py | 6 +++--- 1 file changed, 3 insertions(+), 3 deletions(-) diff --git a/src/backend/base/langflow/base/models/aiml_constants.py b/src/backend/base/langflow/base/models/aiml_constants.py index 5af3b3b4c40b..48fa89ce4c81 100644 --- a/src/backend/base/langflow/base/models/aiml_constants.py +++ b/src/backend/base/langflow/base/models/aiml_constants.py @@ -238,8 +238,8 @@ "wavymulder/Analog-Diffusion", "zero-one-ai/Yi-34B", "zero-one-ai/Yi-34B-Chat", - "zero-one-ai/Yi-6B" - ] + "zero-one-ai/Yi-6B", +] AIML_EMBEDDING_MODELS = [ "text-embedding-3-large", @@ -249,4 +249,4 @@ "textembedding-gecko-multilingual@001", "textembedding-gecko@001", "textembedding-gecko@003", - ] \ No newline at end of file +]