From a1caac9bfbb469256151cf163086b81fecc5817f Mon Sep 17 00:00:00 2001
From: Louis Dureuil <louis@meilisearch.com>
Date: Wed, 7 Feb 2024 14:22:13 +0100
Subject: [PATCH] Correct distribution shifts for new 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 f608f1d12..104decb66 100644
--- a/milli/src/vector/openai.rs
+++ b/milli/src/vector/openai.rs
@@ -96,10 +96,10 @@ impl EmbeddingModel {
                 Some(DistributionShift { current_mean: 0.90, current_sigma: 0.08 })
             }
             EmbeddingModel::TextEmbedding3Large => {
-                Some(DistributionShift { current_mean: 0.90, current_sigma: 0.08 })
+                Some(DistributionShift { current_mean: 0.70, current_sigma: 0.1 })
             }
             EmbeddingModel::TextEmbedding3Small => {
-                Some(DistributionShift { current_mean: 0.90, current_sigma: 0.08 })
+                Some(DistributionShift { current_mean: 0.75, current_sigma: 0.1 })
             }
         }
     }