From 55e968c9e0ba0e4bfb716e5827550156a13681b0 Mon Sep 17 00:00:00 2001 From: Tony Kipkemboi Date: Tue, 19 Nov 2024 17:42:54 -0500 Subject: [PATCH] Update CLI Watson supported models + docs (#1628) --- docs/concepts/llms.mdx | 2 ++ src/crewai/cli/constants.py | 23 ++++++++--------------- 2 files changed, 10 insertions(+), 15 deletions(-) diff --git a/docs/concepts/llms.mdx b/docs/concepts/llms.mdx index 5757feca3..5211e6652 100644 --- a/docs/concepts/llms.mdx +++ b/docs/concepts/llms.mdx @@ -47,6 +47,8 @@ researcher: # llm: ollama/llama3:70b # llm: groq/llama-3.2-90b-vision-preview # llm: watsonx/meta-llama/llama-3-1-70b-instruct + # llm: nvidia_nim/meta/llama3-70b-instruct + # llm: sambanova/Meta-Llama-3.1-8B-Instruct # ... ``` diff --git a/src/crewai/cli/constants.py b/src/crewai/cli/constants.py index 4be08fa2a..e13349155 100644 --- a/src/crewai/cli/constants.py +++ b/src/crewai/cli/constants.py @@ -123,21 +123,14 @@ MODELS = { ], "ollama": ["ollama/llama3.1", "ollama/mixtral"], "watson": [ - "watsonx/google/flan-t5-xxl", - "watsonx/google/flan-ul2", - "watsonx/bigscience/mt0-xxl", - "watsonx/eleutherai/gpt-neox-20b", - "watsonx/ibm/mpt-7b-instruct2", - "watsonx/bigcode/starcoder", - "watsonx/meta-llama/llama-2-70b-chat", - "watsonx/meta-llama/llama-2-13b-chat", - "watsonx/ibm/granite-13b-instruct-v1", - "watsonx/ibm/granite-13b-chat-v1", - "watsonx/google/flan-t5-xl", - "watsonx/ibm/granite-13b-chat-v2", - "watsonx/ibm/granite-13b-instruct-v2", - "watsonx/elyza/elyza-japanese-llama-2-7b-instruct", - "watsonx/ibm-mistralai/mixtral-8x7b-instruct-v01-q", + "watsonx/meta-llama/llama-3-1-70b-instruct", + "watsonx/meta-llama/llama-3-1-8b-instruct", + "watsonx/meta-llama/llama-3-2-11b-vision-instruct", + "watsonx/meta-llama/llama-3-2-1b-instruct", + "watsonx/meta-llama/llama-3-2-90b-vision-instruct", + "watsonx/meta-llama/llama-3-405b-instruct", + "watsonx/mistral/mistral-large", + "watsonx/ibm/granite-3-8b-instruct", ], "bedrock": [ "bedrock/anthropic.claude-3-5-sonnet-20240620-v1:0",