[CI] Torch 2.11 flaky test_spec_decode_logprobs and gritlm tests (#45772)
Signed-off-by: Micah Williamson <micah.williamson@amd.com>
Micah Williamson committed
Jun 22, 2026 at 20:26 UTC
fa36f86d77e7fd051f4356f6c48714fb0a0b1abd
2 files changed
+2
-2
tests/models/language/pooling/test_gritlm.py
+1
-1
@@ -12,7 +12,7 @@ from .embed_utils import run_client_embeddings
12
13
MODEL_NAME = "parasail-ai/GritLM-7B-vllm"
14
MAX_MODEL_LEN = 4000
15
-ATOL = 0.002
15
+ATOL = 2.3e-3
16
17
18
def _arr(arr):
tests/v1/sample/test_logprobs.py
+1
-1
@@ -1174,7 +1174,7 @@ def test_spec_decode_logprobs(
1174
assert len(ref_logprobs) == len(spec_logprobs)
1175
for ref_logprob, spec_logprob in zip(ref_logprobs, spec_logprobs):
1176
assert math.isclose(
1177
- ref_logprob.logprob, spec_logprob.logprob, rel_tol=5e-2, abs_tol=1e-1
1177
+ ref_logprob.logprob, spec_logprob.logprob, rel_tol=5e-2, abs_tol=2.5e-1
1178
), (
1179
f"Logprob mismatch: ref={ref_logprob.logprob} "
1180
f"spec={spec_logprob.logprob} "