Improve the output precision of embedding models (#19092)

2025-06-04 19:48:57 +08:00
parent 8711bc5e68
commit 35cf32df30
8 changed files with 69 additions and 28 deletions
--- a/tests/models/language/pooling/test_nomic.py
+++ b/tests/models/language/pooling/test_nomic.py
@@ -9,18 +9,15 @@ from .mteb_utils import mteb_test_embed_models
 MODELS = [
    EmbedModelInfo("nomic-ai/nomic-embed-text-v1",
                   architecture="NomicBertModel",
-                   dtype="float32",
                   enable_test=True),
    EmbedModelInfo("nomic-ai/nomic-embed-text-v1.5",
                   architecture="NomicBertModel",
-                   dtype="float32",
                   enable_test=False),
    EmbedModelInfo("nomic-ai/CodeRankEmbed",
                   architecture="NomicBertModel",
                   enable_test=False),
    EmbedModelInfo("nomic-ai/nomic-embed-text-v2-moe",
                   architecture="NomicBertModel",
-                   dtype="float32",
                   enable_test=True)
 ]