Expose REST embedder to the API

2025-07-26 16:21:07 +00:00 · 2024-03-25 10:05:38 +01:00
parent f87747f4d3
commit a1db342f01
7 changed files with 357 additions and 32 deletions
--- a/milli/src/vector/hf.rs
+++ b/milli/src/vector/hf.rs
@ -194,7 +194,10 @@ impl Embedder {

    pub fn distribution(&self) -> Option<DistributionShift> {
        if self.options.model == "BAAI/bge-base-en-v1.5" {
-            Some(DistributionShift { current_mean: 0.85, current_sigma: 0.1 })
+            Some(DistributionShift {
+                current_mean: ordered_float::OrderedFloat(0.85),
+                current_sigma: ordered_float::OrderedFloat(0.1),
+            })
        } else {
            None
        }