Unverified Commit 13e79fc8 authored by Angela Yi's avatar Angela Yi Committed by GitHub
Browse files

[ci] Update rtol for test_classification (#36556)


Signed-off-by: default avatarangelayi <yiangela7@gmail.com>
Co-authored-by: default avatarRichard Zou <zou3519@users.noreply.github.com>
parent 9d07a3d6
...@@ -45,5 +45,7 @@ def test_models( ...@@ -45,5 +45,7 @@ def test_models(
# half datatype tests in # half datatype tests in
# tests/models/language/pooling/test_embedding.py # tests/models/language/pooling/test_embedding.py
assert torch.allclose( assert torch.allclose(
hf_output, vllm_output, 1e-3 if dtype == "float" else 1e-2 hf_output,
vllm_output,
rtol=2e-3 if dtype == "float" else 1e-2,
) )
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment