[CI] Add PPL test for generation models (#24485)

Signed-off-by: wang.yuqi <noooop@126.com>
2025-09-10 21:16:39 +08:00
parent d6069887c6
commit bd98842c8a
9 changed files with 211 additions and 7 deletions
--- a/tests/models/language/pooling/embed_utils.py
+++ b/tests/models/language/pooling/embed_utils.py
@@ -59,7 +59,7 @@ def correctness_test_embed_models(hf_runner,

    with hf_runner(
            model_info.name,
-            dtype="float32",
+            dtype=model_info.hf_dtype,
            is_sentence_transformer=True,
    ) as hf_model: