Fix default dimensions for models

2024-02-07 11:48:19 +01:00 · 2024-02-07 11:48:19 +01:00 · 32ee05ccef
parent 74c180267e
commit 32ee05ccef
1 changed files with 2 additions and 2 deletions
--- 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,
        }
    }