From 02095c030f5d24d0266e490f4adfedf18e6d9d47 Mon Sep 17 00:00:00 2001 From: Aiden Dai Date: Wed, 3 Apr 2024 11:46:10 +0800 Subject: [PATCH] Add support of Mistral Large --- src/api/models/bedrock.py | 3 ++- 1 file changed, 2 insertions(+), 1 deletion(-) diff --git a/src/api/models/bedrock.py b/src/api/models/bedrock.py index c86d79b..ddb0d01 100644 --- a/src/api/models/bedrock.py +++ b/src/api/models/bedrock.py @@ -49,6 +49,7 @@ SUPPORTED_BEDROCK_MODELS = { "meta.llama2-70b-chat-v1": "Llama 2 Chat 70B", "mistral.mistral-7b-instruct-v0:2": "Mistral 7B Instruct", "mistral.mixtral-8x7b-instruct-v0:1": "Mixtral 8x7B Instruct", + "mistral.mistral-large-2402-v1:0": "Mistral Large", } SUPPORTED_BEDROCK_EMBEDDING_MODELS = { @@ -657,7 +658,7 @@ def get_model(model_id: str) -> BedrockModel: return ClaudeModel() elif model_name in ["Llama 2 Chat 13B", "Llama 2 Chat 70B"]: return Llama2Model() - elif model_name in ["Mistral 7B Instruct", "Mixtral 8x7B Instruct"]: + elif model_name in ["Mistral 7B Instruct", "Mixtral 8x7B Instruct", "Mistral Large"]: return MistralModel() else: logger.error("Unsupported model id " + model_id)