[New Model]: nomic-embed-text-v2-moe (#17785)

2025-05-11 15:59:43 +08:00
parent 06c0922a69
commit e4b8713380
9 changed files with 899 additions and 364 deletions
--- a/tests/models/utils.py
+++ b/tests/models/utils.py
@@ -332,9 +332,10 @@ def matryoshka_fy(tensor: torch.Tensor, dimensions: int):

 class EmbedModelInfo(NamedTuple):
    name: str
-    is_matryoshka: bool
+    is_matryoshka: bool = False
    matryoshka_dimensions: Optional[list[int]] = None
    architecture: str = ""
+    dtype: str = "auto"
    enable_test: bool = True