[Model] Support is_causal HF config field for Qwen2 model (#10621)

Signed-off-by: DarkLight1337 <tlleungac@connect.ust.hk>
2024-11-25 17:51:20 +08:00
parent 05d1f8c9c6
commit ed46f14321
5 changed files with 51 additions and 13 deletions
--- a/tests/models/embedding/utils.py
+++ b/tests/models/embedding/utils.py
@@ -24,7 +24,7 @@ def check_embeddings_close(
                                  dim=0)

        fail_msg = (f"Test{prompt_idx}:"
-                    f"\n{name_0}:\t{embeddings_0!r}"
-                    f"\n{name_1}:\t{embeddings_1!r}")
+                    f"\n{name_0}:\t{embeddings_0[:16]!r}"
+                    f"\n{name_1}:\t{embeddings_1[:16]!r}")

        assert sim >= 1 - tol, fail_msg