From 32ee05cceffc4cb3425283be67adc8a94d50824c Mon Sep 17 00:00:00 2001 From: Louis Dureuil Date: Wed, 7 Feb 2024 11:48:19 +0100 Subject: [PATCH] Fix default dimensions for models --- milli/src/vector/openai.rs | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) diff --git a/milli/src/vector/openai.rs b/milli/src/vector/openai.rs index 9deb2e2da..f608f1d12 100644 --- a/milli/src/vector/openai.rs +++ b/milli/src/vector/openai.rs @@ -68,8 +68,8 @@ impl EmbeddingModel { pub fn default_dimensions(&self) -> usize { match self { EmbeddingModel::TextEmbeddingAda002 => 1536, - EmbeddingModel::TextEmbedding3Large => 1536, - EmbeddingModel::TextEmbedding3Small => 3072, + EmbeddingModel::TextEmbedding3Large => 3072, + EmbeddingModel::TextEmbedding3Small => 1536, } }