From d5bf7853ffd95ad586498747203f36a575761812 Mon Sep 17 00:00:00 2001
From: Grisha <skvrd@users.noreply.github.com>
Date: Mon, 11 Mar 2024 10:30:06 -0700
Subject: [PATCH] add support for open-mixtral-8x7b (#11792)
---
.../llama_index/llms/mistralai/utils.py | 5 +++++
1 file changed, 5 insertions(+)
diff --git a/llama-index-integrations/llms/llama-index-llms-mistralai/llama_index/llms/mistralai/utils.py b/llama-index-integrations/llms/llama-index-llms-mistralai/llama_index/llms/mistralai/utils.py
index 0201363cb6..5598066d7c 100644
--- a/llama-index-integrations/llms/llama-index-llms-mistralai/llama_index/llms/mistralai/utils.py
+++ b/llama-index-integrations/llms/llama-index-llms-mistralai/llama_index/llms/mistralai/utils.py
@@ -5,6 +5,11 @@ MISTRALAI_MODELS: Dict[str, int] = {
"mistral-small": 32000,
"mistral-medium": 32000,
"mistral-large": 32000,
+ "open-mixtral-8x7b": 32000,
+ "open-mistral-7b": 32000,
+ "mistral-small-latest": 32000,
+ "mistral-medium-latest": 32000,
+ "mistral-large-latest": 32000,
}
--
GitLab