Correct distribution shifts for new models

This commit is contained in:
Louis Dureuil 2024-02-07 14:22:13 +01:00
parent 88d03c56ab
commit a1caac9bfb
No known key found for this signature in database

View File

@ -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 })
}
}
}