[CI/Build] Fix model nightly tests (#26466)

Signed-off-by: DarkLight1337 <tlleungac@connect.ust.hk>
2025-10-09 14:44:16 +08:00
parent d24cf322e1
commit 0f29dca988
5 changed files with 9 additions and 7 deletions
--- a/tests/models/language/pooling/test_token_classification.py
+++ b/tests/models/language/pooling/test_token_classification.py
@@ -67,4 +67,4 @@ def test_modernbert_models(
    for hf_output, vllm_output in zip(hf_outputs, vllm_outputs):
        hf_output = torch.tensor(hf_output).cpu().float()
        vllm_output = torch.tensor(vllm_output).cpu().float()
-        assert torch.allclose(hf_output, vllm_output, 1e-2)
+        assert torch.allclose(hf_output, vllm_output, atol=1e-2)