diff --git a/llama-index-integrations/llms/llama-index-llms-watsonx/llama_index/llms/watsonx/utils.py b/llama-index-integrations/llms/llama-index-llms-watsonx/llama_index/llms/watsonx/utils.py index c19b17efbc5ab..61407d7dd9d3f 100644 --- a/llama-index-integrations/llms/llama-index-llms-watsonx/llama_index/llms/watsonx/utils.py +++ b/llama-index-integrations/llms/llama-index-llms-watsonx/llama_index/llms/watsonx/utils.py @@ -6,17 +6,23 @@ "google/flan-t5-xxl": 4096, "google/flan-ul2": 4096, "bigscience/mt0-xxl": 4096, - "eleutherai/gpt-neox-20b": 8192, - "bigcode/starcoder": 8192, - "codellama/codellama-34b-instruct-hf": 4096, + "bigcode/starcoder": 8192, # deprecated, removed after 25.04.2024 + "core42/jais-13b-chat": 8192, + "codellama/codellama-34b-instruct-hf": 16384, "meta-llama/llama-2-13b-chat": 4096, "meta-llama/llama-2-70b-chat": 4096, - "ibm-mistralai/mixtral-8x7b-instruct-v01-q": 32768, - "ibm/granite-13b-instruct-v1": 8192, - "ibm/granite-13b-chat-v1": 8192, - "ibm/granite-13b-instruct-v2": 8192, + "meta-llama/llama-3-8b-instruct": 8192, + "meta-llama/llama-3-70b-instruct": 8192, + "ibm-mistralai/merlinite-7b": 32768, + "ibm-mistralai/mixtral-8x7b-instruct-v01-q": 32768, # deprecated, removed after 23.05.2024 "ibm/granite-13b-chat-v2": 8192, - "ibm/granite-20b-multilingual": 8190, + "ibm/granite-13b-instruct-v2": 8192, + "ibm/granite-20b-multilingual": 8192, + "ibm/granite-7b-lab": 8192, + "ibm/granite-8b-japanese": 8192, + "mistralai/mixtral-8x7b-instruct-v01": 32768, + "elyza/elyza-japanese-llama-2-7b-instruct": 4096, + "mncai/llama2-13b-dpo-v7": 4096, } diff --git a/llama-index-integrations/llms/llama-index-llms-watsonx/pyproject.toml b/llama-index-integrations/llms/llama-index-llms-watsonx/pyproject.toml index 9fc41a5bcfdbb..e32c381ed2a51 100644 --- a/llama-index-integrations/llms/llama-index-llms-watsonx/pyproject.toml +++ b/llama-index-integrations/llms/llama-index-llms-watsonx/pyproject.toml @@ -27,7 +27,7 @@ exclude = ["**/BUILD"] license = "MIT" name = "llama-index-llms-watsonx" readme = "README.md" -version = "0.1.6" +version = "0.1.7" [tool.poetry.dependencies] python = ">=3.10,<4.0"