mistral:text for embeddings

jupyterlab · Jun 20, 2024 · e28c0c0 · e28c0c0
1 parent 4952a03
commit e28c0c0
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/packages/jupyter-ai-magics/jupyter_ai_magics/embedding_providers.py b/packages/jupyter-ai-magics/jupyter_ai_magics/embedding_providers.py
@@ -74,7 +74,7 @@ class OllamaEmbeddingsProvider(BaseEmbeddingsProvider, OllamaEmbeddings):
     name = "Ollama"
     models = [
         "llama2",
-        "mistral",
+        "mistral:text",
         "tinyllama",
     ]
     model_id_key = "model"